from typing import Any, Dict, Optional

from pydantic import BaseModel, ConfigDict, Field


class AgentInput(BaseModel):
    """统一工作流输入模型（新版本 DTO）"""

    model_config = ConfigDict(extra="forbid")

    query: str
    conversation_id: Optional[str] = None
    workflow_type: str = "conversation"
    response_mode: str = "blocking"
    user: Optional[str] = None
    inputs: Dict[str, Any] = Field(default_factory=dict)