init

schemas/agent_input.py

@@ -1,9 +1,16 @@
-from typing import Optional
-from pydantic import BaseModel
+from typing import Any, Dict, Optional
+
+from pydantic import BaseModel, ConfigDict, Field
 
 
 class AgentInput(BaseModel):
-    """Agent 输入模型"""
-    input: str
-    session_id: Optional[str] = None
+    """统一工作流输入模型（新版本 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)