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="基于现病史、既往史与主诉，针对选定任务的具体指导建议"
    )