39 lines
939 B
Python
39 lines
939 B
Python
|
from pydantic import Field
|
|||
|
|
from agent_system.base import BaseResponseModel
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TaskInfo(BaseResponseModel):
|
|||
|
|
"""
|
|||
|
|
任务信息模型
|
|||
|
|
"""
|
|||
|
|
task_name: str = Field(
|
|||
|
|
...,
|
|||
|
|
description="任务名称"
|
|||
|
|
)
|
|||
|
|
priority: str = Field(
|
|||
|
|
...,
|
|||
|
|
description="任务优先级（紧急、高、中、低）"
|
|||
|
|
)
|
|||
|
|
description: str = Field(
|
|||
|
|
...,
|
|||
|
|
description="任务描述"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ControllerDecision(BaseResponseModel):
|
|||
|
|
"""
|
|||
|
|
Controller智能体决策结果模型
|
|||
|
|
|
|||
|
|
基于未完成的任务列表、现病史、既往史与主诉，
|
|||
|
|
输出选择的任务以及具体的指导建议。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
selected_task: TaskInfo = Field(
|
|||
|
|
...,
|
|||
|
|
description="选择执行的任务信息"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
specific_guidance: str = Field(
|
|||
|
|
...,
|
|||
|
|
description="基于现病史、既往史与主诉，针对选定任务的具体指导建议"
|
|||
|
|
)
|