97 lines
4.3 KiB
Python
97 lines
4.3 KiB
Python
from django.db import models
|
|
from apps.common.models import BaseModel
|
|
from apps.user.models import User
|
|
from apps.case.models import CaseBase
|
|
|
|
|
|
class TrainingRecord(BaseModel):
|
|
"""训练记录表"""
|
|
TRAINING_MODE_CHOICES = [
|
|
('novice', '新手'),
|
|
('practice', '练习'),
|
|
('exam', '考试'),
|
|
]
|
|
EVALUATION_LEVEL_CHOICES = [
|
|
('excellent', '优秀'),
|
|
('good', '良好'),
|
|
('average', '一般'),
|
|
('poor', '较差'),
|
|
]
|
|
STATUS_CHOICES = [
|
|
('in_progress', '进行中'),
|
|
('completed', '已完成'),
|
|
('aborted', '已中断'),
|
|
]
|
|
|
|
id = models.BigAutoField(primary_key=True)
|
|
user = models.ForeignKey(
|
|
User, on_delete=models.CASCADE,
|
|
related_name='training_records', verbose_name='用户'
|
|
)
|
|
case = models.ForeignKey(
|
|
CaseBase, on_delete=models.CASCADE,
|
|
related_name='training_records', verbose_name='病例'
|
|
)
|
|
training_mode = models.CharField('训练模式', max_length=50, choices=TRAINING_MODE_CHOICES)
|
|
case_type = models.CharField('病例类型', max_length=30, blank=True)
|
|
teacher = models.ForeignKey(
|
|
User, on_delete=models.SET_NULL,
|
|
null=True, blank=True, related_name='supervised_records',
|
|
verbose_name='带教老师'
|
|
)
|
|
start_time = models.DateTimeField('开始时间', auto_now_add=True)
|
|
end_time = models.DateTimeField('结束时间', null=True, blank=True)
|
|
duration_seconds = models.IntegerField('训练时长', null=True, blank=True)
|
|
total_score = models.DecimalField('总分', max_digits=5, decimal_places=2, null=True, blank=True)
|
|
ai_score = models.DecimalField('AI评分', max_digits=5, decimal_places=2, null=True, blank=True)
|
|
teacher_score = models.DecimalField('教师评分', max_digits=5, decimal_places=2, null=True, blank=True)
|
|
evaluation_level = models.CharField('评价等级', max_length=20, choices=EVALUATION_LEVEL_CHOICES, blank=True)
|
|
status = models.CharField('状态', max_length=30, choices=STATUS_CHOICES, default='in_progress')
|
|
feedback = models.TextField('总评', blank=True)
|
|
thinking_chain = models.TextField('临床推理链', blank=True)
|
|
diagnosis_path = models.TextField('诊断路径', blank=True)
|
|
wrong_points = models.JSONField('错误知识点', default=list, blank=True)
|
|
missed_questions = models.JSONField('漏问项', default=list, blank=True)
|
|
recommendation_result = models.JSONField('AI推荐', default=dict, blank=True)
|
|
ai_feedback_structured = models.JSONField('AI结构化反馈', default=dict, blank=True)
|
|
osce_station_score = models.JSONField('OSCE各站点成绩', default=dict, blank=True)
|
|
interruption_count = models.IntegerField('中断次数', default=0)
|
|
emotion_analysis = models.JSONField('情绪分析', default=dict, blank=True)
|
|
prompt_version = models.CharField('Prompt版本', max_length=50, blank=True)
|
|
rag_context_version = models.CharField('知识上下文版本', max_length=50, blank=True)
|
|
|
|
class Meta:
|
|
db_table = 'training_record'
|
|
verbose_name = '训练记录'
|
|
verbose_name_plural = '训练记录'
|
|
|
|
def __str__(self):
|
|
return f"{self.user.username} - {self.case.title}"
|
|
|
|
|
|
class TrainingScoreDetail(BaseModel):
|
|
"""评分明细表"""
|
|
id = models.BigAutoField(primary_key=True)
|
|
record = models.ForeignKey(
|
|
TrainingRecord, on_delete=models.CASCADE,
|
|
related_name='score_details', verbose_name='训练记录'
|
|
)
|
|
rule = models.ForeignKey(
|
|
'case.ScoringRule', on_delete=models.CASCADE,
|
|
null=True, blank=True, verbose_name='评分规则'
|
|
)
|
|
dimension = models.CharField('评分维度', max_length=50)
|
|
score = models.DecimalField('分数', max_digits=5, decimal_places=2)
|
|
deducted_reason = models.TextField('扣分原因', blank=True)
|
|
evidence_message_ids = models.JSONField('对应对话证据', default=list, blank=True)
|
|
ai_confidence = models.DecimalField('AI评分置信度', max_digits=5, decimal_places=2, null=True, blank=True)
|
|
comment = models.TextField('评语', blank=True)
|
|
|
|
class Meta:
|
|
db_table = 'training_score_detail'
|
|
verbose_name = '评分明细'
|
|
verbose_name_plural = '评分明细'
|
|
|
|
def __str__(self):
|
|
return f"{self.record} - {self.dimension}: {self.score}"
|