刘金宝
37 lines
1004 B
Python
37 lines
1004 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class ConfigOption(BaseModel):
|
|
"""训练配置选项：用于前端渲染单个可选项。"""
|
|
|
|
value: str
|
|
label: str
|
|
description: str | None = None
|
|
|
|
|
|
class PatientConfig(BaseModel):
|
|
"""病人初始化配置：控制 AI 病人的就诊场景、年龄段、文化程度和性格。"""
|
|
|
|
visit_environment: str = "outpatient"
|
|
age_group: str = "youth"
|
|
education_level: str = "higher"
|
|
personality: str = "calm"
|
|
|
|
|
|
class TrainingConfigOptionsResponse(BaseModel):
|
|
"""训练配置响应：返回默认配置和全部可选项。"""
|
|
|
|
case_id: int
|
|
recommended: PatientConfig
|
|
recommended_labels: dict[str, str]
|
|
options: dict[str, list[ConfigOption]]
|
|
|
|
|
|
class TrainingConfigRecommendedResponse(BaseModel):
|
|
"""推荐训练配置响应：用于训练页进入时初始化默认病人信息。"""
|
|
|
|
case_id: int
|
|
recommended: PatientConfig
|
|
recommended_labels: dict[str, str]
|
|
options: dict[str, list[ConfigOption]]
|