This commit is contained in:
2026-03-02 17:41:04 +08:00
parent 460c2e87b8
commit c9bebe5615
7 changed files with 322 additions and 86 deletions
+12 -5
View File
@@ -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)
+22
View File
@@ -0,0 +1,22 @@
from typing import Any, Dict, List, Optional
from pydantic import BaseModel, ConfigDict, Field
class ChatMessageFileDTO(BaseModel):
"""文件输入模型(预留)"""
model_config = ConfigDict(extra="allow")
class ChatMessageRequestDTO(BaseModel):
"""/api/workflows/stream 请求模型(对齐标准 chat message 请求)"""
model_config = ConfigDict(extra="forbid")
query: str
inputs: Dict[str, Any] = Field(default_factory=dict)
response_mode: str = "streaming"
user: Optional[str] = None
conversation_id: Optional[str] = None
files: List[ChatMessageFileDTO] = Field(default_factory=list)
+7
View File
@@ -0,0 +1,7 @@
from schemas.agent_input import AgentInput
class StreamInputDTO(AgentInput):
"""流式输入 DTO(与统一 AgentInput 保持一致)"""
response_mode: str = "streaming"