tingwu_new

backend/ai_services/serializers.py

@@ -1,11 +1,18 @@
 from rest_framework import serializers
-from .models import TranscriptionTask
+from .models import TranscriptionTask, AIEvaluation
+
+class AIEvaluationSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = AIEvaluation
+        fields = ['id', 'score', 'evaluation', 'model_selection', 'prompt', 'reasoning', 'status', 'error_message', 'created_at', 'updated_at']
 
 class TranscriptionTaskSerializer(serializers.ModelSerializer):
+    ai_evaluation = AIEvaluationSerializer(read_only=True)
+
     class Meta:
         model = TranscriptionTask
-        fields = ['id', 'file_url', 'task_id', 'status', 'transcription', 'summary', 'error_message', 'created_at', 'updated_at', 'score', 'evaluation', 'transcription_data', 'summary_data', 'auto_chapters_data']
-        read_only_fields = ['id', 'file_url', 'task_id', 'status', 'transcription', 'summary', 'error_message', 'created_at', 'updated_at', 'score', 'evaluation', 'transcription_data', 'summary_data', 'auto_chapters_data']
+        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']
 
 class TranscriptionUploadSerializer(serializers.Serializer):
     file = serializers.FileField(help_text="上传的音频文件")