add training configuration APIs
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
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]]
|
||||
Reference in New Issue
Block a user