init medical training project

2026-05-29 15:58:00 +08:00
commit b4bb38b7be
91 changed files with 6765 additions and 0 deletions
@@ -0,0 +1,96 @@
+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}"