31 lines
993 B
Python
31 lines
993 B
Python
from django.contrib import admin
|
|
from django.db import models
|
|
from django.forms import Textarea
|
|
|
|
from .models import ChatConversation, ChatMessage, ChatSystemPrompt
|
|
|
|
|
|
@admin.register(ChatConversation)
|
|
class ChatConversationAdmin(admin.ModelAdmin):
|
|
list_display = ("id", "user", "title", "updated_at", "created_at")
|
|
search_fields = ("title", "user__username")
|
|
list_filter = ("created_at", "updated_at")
|
|
|
|
|
|
@admin.register(ChatMessage)
|
|
class ChatMessageAdmin(admin.ModelAdmin):
|
|
list_display = ("id", "conversation", "role", "name", "created_at")
|
|
search_fields = ("conversation__id", "content", "name")
|
|
list_filter = ("role", "created_at")
|
|
|
|
|
|
@admin.register(ChatSystemPrompt)
|
|
class ChatSystemPromptAdmin(admin.ModelAdmin):
|
|
readonly_fields = ("updated_at",)
|
|
formfield_overrides = {
|
|
models.TextField: {"widget": Textarea(attrs={"rows": 20, "cols": 120})},
|
|
}
|
|
|
|
def has_add_permission(self, request):
|
|
return not ChatSystemPrompt.objects.exists()
|