from typing import Optional
from pydantic import BaseModel


class AgentInput(BaseModel):
    """Agent 输入模型"""
    input: str
    session_id: Optional[str] = None
    workflow_type: str = "conversation"