tingwu_new

backend/ai_services/models.py

@@ -25,8 +25,11 @@ class TranscriptionTask(models.Model):
     
     transcription = models.TextField(verbose_name=_('逐字稿'), blank=True, null=True)
     summary = models.TextField(verbose_name=_('AI总结'), blank=True, null=True)
-    score = models.IntegerField(verbose_name=_('AI评分'), blank=True, null=True, help_text=_('基于转写内容的评分'))
-    evaluation = models.TextField(verbose_name=_('AI评语'), blank=True, null=True)
+    
+    # 已解耦到 AIEvaluation 模型
+    # score = models.IntegerField(verbose_name=_('AI评分'), blank=True, null=True, help_text=_('基于转写内容的评分'))
+    # evaluation = models.TextField(verbose_name=_('AI评语'), blank=True, null=True)
+    
     error_message = models.TextField(verbose_name=_('错误信息'), blank=True, null=True)
     created_at = models.DateTimeField(verbose_name=_('创建时间'), auto_now_add=True)
     updated_at = models.DateTimeField(verbose_name=_('更新时间'), auto_now=True)
@@ -38,3 +41,58 @@ class TranscriptionTask(models.Model):
 
     def __str__(self):
         return f"{self.id} - {self.get_status_display()}"
+
+
+class AIEvaluation(models.Model):
+    class Status(models.TextChoices):
+        PENDING = 'PENDING', _('等待中')
+        PROCESSING = 'PROCESSING', _('生成中')
+        COMPLETED = 'COMPLETED', _('已完成')
+        FAILED = 'FAILED', _('失败')
+
+    task = models.OneToOneField(
+        TranscriptionTask, 
+        on_delete=models.CASCADE, 
+        related_name='ai_evaluation',
+        verbose_name=_('关联任务')
+    )
+    
+    # 评分与评语
+    score = models.IntegerField(verbose_name=_('AI评分'), blank=True, null=True, help_text=_('0-100分'))
+    evaluation = models.TextField(verbose_name=_('AI评语'), blank=True, null=True)
+    
+    # 配置选项 (可在Admin中设置)
+    model_selection = models.CharField(
+        verbose_name=_('模型选择'), 
+        max_length=50, 
+        default='qwen-plus', 
+        help_text=_('例如: qwen-plus, qwen-turbo, qwen-max')
+    )
+    prompt = models.TextField(
+        verbose_name=_('评分提示词'), 
+        default='你是一个专业的评分助手。请根据提供的转写内容，对内容质量、逻辑清晰度、语言表达等方面进行综合评分（0-100分），并给出详细的评语。请以JSON格式返回，包含"score"和"evaluation"字段。',
+        help_text=_('用于指导AI评分的提示词')
+    )
+    
+    # 原始数据与推理
+    raw_response = models.JSONField(verbose_name=_('原始响应'), blank=True, null=True, help_text=_('大模型返回的完整JSON'))
+    reasoning = models.TextField(verbose_name=_('推理过程'), blank=True, null=True, help_text=_('AI的推理过程（如果有）'))
+    
+    status = models.CharField(
+        verbose_name=_('评估状态'), 
+        max_length=20, 
+        choices=Status.choices, 
+        default=Status.PENDING
+    )
+    error_message = models.TextField(verbose_name=_('错误信息'), blank=True, null=True)
+    
+    created_at = models.DateTimeField(verbose_name=_('创建时间'), auto_now_add=True)
+    updated_at = models.DateTimeField(verbose_name=_('更新时间'), auto_now=True)
+
+    class Meta:
+        verbose_name = _('AI智能评估')
+        verbose_name_plural = _('AI智能评估')
+        ordering = ['-created_at']
+
+    def __str__(self):
+        return f"Evaluation for Task {self.task.id}"