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")}),
    )