tingwu_new

backend/ai_services/admin.py

@@ -1,16 +1,16 @@
 from django.contrib import admin
 from unfold.admin import ModelAdmin as UnfoldModelAdmin
 from unfold.admin import StackedInline as UnfoldStackedInline
-from .models import TranscriptionTask, AIEvaluation
+from .models import TranscriptionTask, AIEvaluation, AIEvaluationTemplate
 
 class AIEvaluationInline(UnfoldStackedInline):
     model = AIEvaluation
     extra = 0
-    can_delete = False
-    verbose_name = "AI评估"
-    verbose_name_plural = "AI评估"
-    readonly_fields = ['created_at', 'updated_at', 'raw_response', 'reasoning']
-    fields = ('score', 'evaluation', 'model_selection', 'prompt', 'reasoning', 'status', 'error_message')
+    can_delete = True
+    verbose_name = "AI评估结果"
+    verbose_name_plural = "AI评估结果"
+    readonly_fields = ['created_at', 'updated_at', 'raw_response', 'reasoning', 'template']
+    fields = ('template', 'score', 'evaluation', 'model_selection', 'prompt', 'reasoning', 'status', 'error_message')
 
 @admin.register(TranscriptionTask)
 class TranscriptionTaskAdmin(UnfoldModelAdmin):
@@ -20,17 +20,23 @@ class TranscriptionTaskAdmin(UnfoldModelAdmin):
     readonly_fields = ['id', 'created_at', 'updated_at', 'task_id']
     inlines = [AIEvaluationInline]
 
+@admin.register(AIEvaluationTemplate)
+class AIEvaluationTemplateAdmin(UnfoldModelAdmin):
+    list_display = ['name', 'model_selection', 'is_active', 'created_at']
+    list_filter = ['is_active', 'model_selection', 'created_at']
+    search_fields = ['name', 'prompt']
+
 @admin.register(AIEvaluation)
 class AIEvaluationAdmin(UnfoldModelAdmin):
-    list_display = ['id', 'task', 'score', 'status', 'model_selection', 'created_at']
-    list_filter = ['status', 'model_selection', 'created_at']
+    list_display = ['id', 'task', 'template', 'score', 'status', 'model_selection', 'created_at']
+    list_filter = ['status', 'model_selection', 'created_at', 'template']
     search_fields = ['task__id', 'evaluation', 'reasoning']
     readonly_fields = ['id', 'created_at', 'updated_at', 'raw_response']
     fieldsets = (
         (None, {
-            'fields': ('task', 'status', 'score', 'evaluation')
+            'fields': ('task', 'template', 'status', 'score', 'evaluation')
         }),
-        ('配置', {
+        ('配置快照', {
             'fields': ('model_selection', 'prompt'),
             'classes': ('collapse',),
         }),