24 lines
948 B
Python
24 lines
948 B
Python
|
|
from django.contrib import admin
|
|
from .models import AIInteraction, AgentConfig
|
|
|
|
|
|
@admin.register(AIInteraction)
|
|
class AIInteractionAdmin(admin.ModelAdmin):
|
|
list_display = ["user", "message_role", "tokens_used", "created_at"]
|
|
list_filter = ["message_role"]
|
|
|
|
|
|
@admin.register(AgentConfig)
|
|
class AgentConfigAdmin(admin.ModelAdmin):
|
|
list_display = ["name", "provider", "model_name", "is_active", "is_default", "created_at"]
|
|
list_filter = ["provider", "is_active", "is_default"]
|
|
search_fields = ["name"]
|
|
fieldsets = (
|
|
(None, {"fields": ("name", "description", "provider")}),
|
|
("API Configuration", {"fields": ("api_key", "api_secret", "base_url")}),
|
|
("Model", {"fields": ("model_name", "temperature", "max_tokens")}),
|
|
("Prompt", {"fields": ("system_prompt",)}),
|
|
("Capabilities", {"fields": ("capabilities",)}),
|
|
("Settings", {"fields": ("is_active", "is_default", "user")}),
|
|
)
|