feat: update cms case api

2026-06-12 17:19:23 +08:00
parent 2fab2be0a1
commit 8fecaeeb54
14 changed files with 1375 additions and 237 deletions
@@ -72,6 +72,54 @@ def parse_pdf(files, case_type: str, user) -> dict:
    }


+def generate_from_prompt(prompt: str, case_type: str, user) -> dict:
+    """CMS-CASE-AI-1: 一段长 prompt → DeepSeek 按病例模板生成结构化病例内容。
+
+    产出与 parse_pdf 同构（不落库、不含评分规则），缓存 parse_id 供后续 full-create 复用。
+    """
+    if case_type not in ('traditional', 'teaching'):
+        raise AppError('CASE_TYPE_NOT_SUPPORTED', f'case_type 不支持: {case_type}', status_code=400)
+    if not (prompt or '').strip():
+        raise AppError('CASE_VALIDATION_ERROR', 'prompt 必填', status_code=400)
+
+    t0 = time.time()
+
+    prompt_name = f'case_{case_type}_full'
+    system_prompt, prompt_version = load_prompt(prompt_name)
+
+    result = deepseek_client.call_deepseek(system_prompt, prompt)
+    data = result['data']
+
+    data.pop('scoring_rules', None)
+    data.pop('stages', None)
+
+    data['case_type'] = case_type
+    if 'exam_items' in data:
+        data['exam_items'] = normalize_exam_items(data.get('exam_items') or [])
+    else:
+        data['exam_items'] = []
+
+    _strip_unknown_fields(data)
+    _validate_schema(data)
+
+    parse_id = uuid.uuid4().hex[:12]
+    cache.set(f'parse_result:{parse_id}', json.dumps(data, ensure_ascii=False), PARSE_RESULT_TTL)
+
+    audit.info(
+        'CASE_AI_GENERATE user=%s parse_id=%s tokens=%s prompt_version=%s',
+        user.id, parse_id, result.get('usage', {}), prompt_version,
+    )
+
+    return {
+        'parse_id': parse_id,
+        'case_type': case_type,
+        'ai_usage': result.get('usage', {}),
+        'prompt_version': prompt_version,
+        'generating_seconds': round(time.time() - t0, 1),
+        'data': data,
+    }
+
+
 _SCHEMA_ALLOWED_KEYS = {
    'title', 'case_type', 'difficulty', 'chief_complaint', 'description',
    'patient_age', 'patient_gender', 'tags', 'symptom_tags', 'disease_tags',
@@ -0,0 +1,216 @@
+"""病例落库 / 完整结构组装的共享逻辑。
+
+`/api/case/cases/full-create`（业务域）与 `/api/cms/cases/`（CMS 内容管理）共用同一套
+创建与序列化逻辑，避免两处实现漂移。
+"""
+from django.db import transaction
+
+from config.exceptions import AppError
+from ..models import (
+    CaseBase, TraditionalCase, TeachingCase, ScoringRule, CaseExamItem,
+)
+from .department_resolver import resolve_department
+from .exam_items import (
+    normalize_exam_items, assert_no_duplicate_exam_items, EXAM_ITEM_FIELDS,
+)
+
+TRADITIONAL_FIELDS = {'standard_diagnosis', 'standard_treatment', 'guideline_reference'}
+TEACHING_FIELDS = {'teaching_goal', 'discussion_questions', 'teacher_guide', 'scoring_focus'}
+SCORING_RULE_FIELDS = {
+    'dimension', 'competency_dimension', 'score_weight',
+    'ai_auto_score', 'osce_dimension', 'scoring_standard', 'rubric_json',
+}
+CASE_BASE_FIELDS = {
+    'title', 'case_type', 'difficulty', 'difficulty_score',
+    'chief_complaint', 'description', 'patient_age', 'patient_gender',
+    'tags', 'symptom_tags', 'disease_tags', 'competency_tags',
+    'guideline_tags', 'knowledge_points', 'icd_codes',
+    'estimated_minutes', 'osce_enabled', 'rag_enabled',
+    'ai_prompt_template', 'multimodal_assets',
+}
+
+
+def validate_scoring_rules(rules):
+    if not isinstance(rules, list):
+        raise AppError('CASE_VALIDATION_ERROR', 'scoring_rules 必须为数组', status_code=400)
+    for i, rule in enumerate(rules):
+        if not isinstance(rule, dict):
+            raise AppError('CASE_VALIDATION_ERROR', f'scoring_rules[{i}] 必须为对象', status_code=400)
+        if not rule.get('dimension'):
+            raise AppError('CASE_VALIDATION_ERROR', f'scoring_rules[{i}].dimension 必填', status_code=400)
+        weight = rule.get('score_weight')
+        if weight is not None:
+            try:
+                weight = float(weight)
+            except (TypeError, ValueError):
+                raise AppError('CASE_VALIDATION_ERROR', f'scoring_rules[{i}].score_weight 须为数字', status_code=400)
+            if weight <= 0 or weight > 1:
+                raise AppError('CASE_VALIDATION_ERROR', f'scoring_rules[{i}].score_weight 须在 (0, 1]', status_code=400)
+
+
+_UNSET = object()
+
+
+def create_case_full(data, user, institution=_UNSET):
+    """主表 + 子表 + scoring_rules(≥1) + exam_items 同一事务入库。
+
+    institution：病例所属机构；默认取创建者所属机构（`user.institution`）。
+    返回 (case, scoring_rule_count, exam_item_count)。校验失败抛 AppError。
+    """
+    if institution is _UNSET:
+        institution = getattr(user, 'institution', None)
+    case_type = data.get('case_type', '')
+    if case_type not in ('traditional', 'teaching'):
+        raise AppError('CASE_TYPE_NOT_SUPPORTED', f'case_type 不支持: {case_type}', status_code=400)
+
+    if 'stages' in data:
+        raise AppError('CASE_FIELD_NOT_ALLOWED', '本接口不接收 stages 字段', status_code=400)
+
+    sub_data = data.get(case_type)
+    other_type = 'teaching' if case_type == 'traditional' else 'traditional'
+    if not sub_data:
+        raise AppError('CASE_SUBTYPE_REQUIRED', f'{case_type} 子表数据缺失', status_code=400)
+    if data.get(other_type):
+        raise AppError('CASE_SUBTYPE_CONFLICT', '不允许同时传入两种子表数据', status_code=400)
+
+    scoring_rules_data = data.get('scoring_rules', [])
+    if not scoring_rules_data:
+        raise AppError('CASE_VALIDATION_ERROR', 'scoring_rules 必填且至少 1 条', status_code=400)
+    validate_scoring_rules(scoring_rules_data)
+
+    exam_items_data = normalize_exam_items(data.get('exam_items') or [])
+    assert_no_duplicate_exam_items(exam_items_data)
+
+    department = resolve_department(data.get('department_name', ''))
+
+    with transaction.atomic():
+        case_kwargs = {k: data[k] for k in CASE_BASE_FIELDS if k in data}
+        case_kwargs['department'] = department
+        case_kwargs['institution'] = institution
+        case_kwargs['created_by'] = user
+        case_kwargs['status'] = 1
+        case_kwargs['vector_status'] = 0
+        case_kwargs['publish_status'] = 1 if data.get('auto_publish') else 0
+        case = CaseBase.objects.create(**case_kwargs)
+
+        sub_model = TraditionalCase if case_type == 'traditional' else TeachingCase
+        allowed_sub = TRADITIONAL_FIELDS if case_type == 'traditional' else TEACHING_FIELDS
+        sub_model.objects.create(case=case, **{k: v for k, v in sub_data.items() if k in allowed_sub})
+
+        rule_objs = [
+            ScoringRule(case=case, **{k: v for k, v in rule.items() if k in SCORING_RULE_FIELDS})
+            for rule in scoring_rules_data
+        ]
+        ScoringRule.objects.bulk_create(rule_objs)
+
+        if exam_items_data:
+            exam_objs = [
+                CaseExamItem(case=case, **{k: v for k, v in item.items() if k in EXAM_ITEM_FIELDS})
+                for item in exam_items_data
+            ]
+            CaseExamItem.objects.bulk_create(exam_objs)
+
+    return case, len(rule_objs), len(exam_items_data)
+
+
+def build_full_response(case):
+    """组装病例完整结构（主表 + 子表 + scoring_rules + exam_items）。"""
+    if hasattr(case, '_prefetched_objects_cache') and 'exam_items' in getattr(
+        case, '_prefetched_objects_cache', {}
+    ):
+        exam_qs = case.exam_items.all()
+    else:
+        exam_qs = CaseExamItem.objects.filter(case_id=case.id).order_by('display_order', 'id')
+
+    result = {
+        'case': {
+            'id': case.id,
+            'title': case.title,
+            'case_type': case.case_type,
+            'difficulty': case.difficulty,
+            'difficulty_score': case.difficulty_score,
+            'institution': case.institution_id,
+            'institution_name': case.institution.name if case.institution else None,
+            'department': case.department_id,
+            'department_name': case.department.name if case.department else None,
+            'chief_complaint': case.chief_complaint,
+            'description': case.description,
+            'patient_age': case.patient_age,
+            'patient_gender': case.patient_gender,
+            'tags': case.tags,
+            'symptom_tags': case.symptom_tags,
+            'disease_tags': case.disease_tags,
+            'competency_tags': case.competency_tags,
+            'guideline_tags': case.guideline_tags,
+            'knowledge_points': case.knowledge_points,
+            'icd_codes': case.icd_codes,
+            'estimated_minutes': case.estimated_minutes,
+            'osce_enabled': case.osce_enabled,
+            'rag_enabled': case.rag_enabled,
+            'ai_prompt_template': case.ai_prompt_template,
+            'multimodal_assets': case.multimodal_assets,
+            'vector_status': case.vector_status,
+            'publish_status': case.publish_status,
+            'status': case.status,
+            'created_by': case.created_by_id,
+            'created_by_name': case.created_by.real_name if case.created_by else None,
+            'created_at': case.created_at.isoformat() if case.created_at else None,
+            'updated_at': case.updated_at.isoformat() if case.updated_at else None,
+        },
+    }
+
+    if case.case_type == 'traditional':
+        try:
+            tc = case.traditionalcase
+            result['traditional'] = {
+                'standard_diagnosis': tc.standard_diagnosis,
+                'standard_treatment': tc.standard_treatment,
+                'guideline_reference': tc.guideline_reference,
+            }
+        except TraditionalCase.DoesNotExist:
+            result['traditional'] = None
+    elif case.case_type == 'teaching':
+        try:
+            tc = case.teachingcase
+            result['teaching'] = {
+                'teaching_goal': tc.teaching_goal,
+                'discussion_questions': tc.discussion_questions,
+                'teacher_guide': tc.teacher_guide,
+                'scoring_focus': tc.scoring_focus,
+            }
+        except TeachingCase.DoesNotExist:
+            result['teaching'] = None
+
+    rules = case.scoring_rules.all().order_by('id')
+    result['scoring_rules'] = [
+        {
+            'id': r.id,
+            'dimension': r.dimension,
+            'competency_dimension': r.competency_dimension,
+            'score_weight': float(r.score_weight),
+            'ai_auto_score': r.ai_auto_score,
+            'osce_dimension': r.osce_dimension,
+            'scoring_standard': r.scoring_standard,
+            'rubric_json': r.rubric_json,
+        }
+        for r in rules
+    ]
+
+    result['exam_items'] = [
+        {
+            'id': e.id,
+            'item_code': e.item_code,
+            'item_name': e.item_name,
+            'item_type': e.item_type,
+            'category': e.category,
+            'result_text': e.result_text,
+            'result_structured': e.result_structured,
+            'is_key': e.is_key,
+            'is_abnormal': e.is_abnormal,
+            'score_weight': float(e.score_weight),
+            'display_order': e.display_order,
+        }
+        for e in exam_qs
+    ]
+
+    return result