151 lines
6.4 KiB
Python
151 lines
6.4 KiB
Python
import uuid
|
||
from django.db import models
|
||
from django.utils.translation import gettext_lazy as _
|
||
|
||
class TranscriptionTask(models.Model):
|
||
class Status(models.TextChoices):
|
||
PENDING = 'PENDING', _('等待中')
|
||
PROCESSING = 'PROCESSING', _('处理中')
|
||
SUCCEEDED = 'SUCCEEDED', _('成功')
|
||
FAILED = 'FAILED', _('失败')
|
||
|
||
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
||
file_url = models.URLField(verbose_name=_('文件链接'), max_length=1024)
|
||
task_id = models.CharField(verbose_name=_('听悟任务ID'), max_length=100, blank=True, null=True)
|
||
status = models.CharField(
|
||
verbose_name=_('状态'),
|
||
max_length=20,
|
||
choices=Status.choices,
|
||
default=Status.PENDING
|
||
)
|
||
# 存储阿里云听悟返回的原始 JSON 结构
|
||
transcription_data = models.JSONField(verbose_name=_('转写原始数据'), blank=True, null=True, help_text=_('阿里云返回的Transcription完整JSON'))
|
||
summary_data = models.JSONField(verbose_name=_('总结原始数据'), blank=True, null=True, help_text=_('阿里云返回的Summarization完整JSON'))
|
||
auto_chapters_data = models.JSONField(verbose_name=_('章节原始数据'), blank=True, null=True, help_text=_('阿里云返回的AutoChapters完整JSON'))
|
||
|
||
project = models.ForeignKey(
|
||
'competition.Project',
|
||
on_delete=models.SET_NULL,
|
||
null=True,
|
||
blank=True,
|
||
related_name='transcription_tasks',
|
||
verbose_name=_('关联参赛项目')
|
||
)
|
||
|
||
transcription = models.TextField(verbose_name=_('逐字稿'), blank=True, null=True)
|
||
summary = 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)
|
||
|
||
class Meta:
|
||
verbose_name = _('转写任务')
|
||
verbose_name_plural = _('转写任务')
|
||
ordering = ['-created_at']
|
||
|
||
def __str__(self):
|
||
return f"{self.id} - {self.get_status_display()}"
|
||
|
||
|
||
class AIEvaluationTemplate(models.Model):
|
||
name = models.CharField(verbose_name=_('模板名称'), max_length=100, help_text=_('例如：销售话术评分、逻辑性分析'))
|
||
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评分的提示词')
|
||
)
|
||
score_dimension = models.ForeignKey(
|
||
'competition.ScoreDimension',
|
||
on_delete=models.SET_NULL,
|
||
null=True,
|
||
blank=True,
|
||
verbose_name=_('关联评分维度'),
|
||
help_text=_('如果同步到比赛评分，优先使用此维度。未填写则默认使用"AI Rating"或包含"AI"的维度')
|
||
)
|
||
is_default = models.BooleanField(
|
||
verbose_name=_('是否为默认模板'),
|
||
default=False,
|
||
help_text=_('默认模板会评价所有比赛，非默认模板且未关联评分维度时不会自动评价')
|
||
)
|
||
is_active = models.BooleanField(verbose_name=_('是否启用'), default=True, help_text=_('启用后，新的转写任务完成后将自动使用此模板进行评估'))
|
||
|
||
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 self.name
|
||
|
||
|
||
class AIEvaluation(models.Model):
|
||
class Status(models.TextChoices):
|
||
PENDING = 'PENDING', _('等待中')
|
||
PROCESSING = 'PROCESSING', _('生成中')
|
||
COMPLETED = 'COMPLETED', _('已完成')
|
||
FAILED = 'FAILED', _('失败')
|
||
|
||
task = models.ForeignKey(
|
||
TranscriptionTask,
|
||
on_delete=models.CASCADE,
|
||
related_name='ai_evaluations',
|
||
verbose_name=_('关联任务')
|
||
)
|
||
template = models.ForeignKey(
|
||
AIEvaluationTemplate,
|
||
on_delete=models.SET_NULL,
|
||
null=True,
|
||
blank=True,
|
||
related_name='evaluations',
|
||
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)
|
||
|
||
# 记录当时的配置 (快照)
|
||
model_selection = models.CharField(
|
||
verbose_name=_('模型选择'),
|
||
max_length=50,
|
||
default='qwen-plus'
|
||
)
|
||
prompt = models.TextField(verbose_name=_('评分提示词'))
|
||
|
||
# 原始数据与推理
|
||
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} ({self.template.name if self.template else 'Custom'})"
|