tingwu_new

backend/ai_services/serializers.py

@@ -1,18 +1,25 @@
 from rest_framework import serializers
-from .models import TranscriptionTask, AIEvaluation
+from .models import TranscriptionTask, AIEvaluation, AIEvaluationTemplate
+
+class AIEvaluationTemplateSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = AIEvaluationTemplate
+        fields = ['id', 'name', 'model_selection', 'prompt', 'is_active', 'created_at']
 
 class AIEvaluationSerializer(serializers.ModelSerializer):
+    template = AIEvaluationTemplateSerializer(read_only=True)
+    
     class Meta:
         model = AIEvaluation
-        fields = ['id', 'score', 'evaluation', 'model_selection', 'prompt', 'reasoning', 'status', 'error_message', 'created_at', 'updated_at']
+        fields = ['id', 'template', 'score', 'evaluation', 'model_selection', 'prompt', 'reasoning', 'status', 'error_message', 'created_at', 'updated_at']
 
 class TranscriptionTaskSerializer(serializers.ModelSerializer):
-    ai_evaluation = AIEvaluationSerializer(read_only=True)
+    ai_evaluations = AIEvaluationSerializer(many=True, read_only=True)
 
     class Meta:
         model = TranscriptionTask
-        fields = ['id', 'file_url', 'task_id', 'status', 'transcription', 'summary', 'error_message', 'created_at', 'updated_at', 'transcription_data', 'summary_data', 'auto_chapters_data', 'ai_evaluation']
-        read_only_fields = ['id', 'file_url', 'task_id', 'status', 'transcription', 'summary', 'error_message', 'created_at', 'updated_at', 'transcription_data', 'summary_data', 'auto_chapters_data', 'ai_evaluation']
+        fields = ['id', 'file_url', 'task_id', 'status', 'transcription', 'summary', 'error_message', 'created_at', 'updated_at', 'transcription_data', 'summary_data', 'auto_chapters_data', 'ai_evaluations']
+        read_only_fields = ['id', 'file_url', 'task_id', 'status', 'transcription', 'summary', 'error_message', 'created_at', 'updated_at', 'transcription_data', 'summary_data', 'auto_chapters_data', 'ai_evaluations']
 
 class TranscriptionUploadSerializer(serializers.Serializer):
     file = serializers.FileField(help_text="上传的音频文件")