新增标题提取
This commit is contained in:
37
src/schemas/deep_agent_chat.py
Normal file
37
src/schemas/deep_agent_chat.py
Normal file
@@ -0,0 +1,37 @@
|
||||
from pydantic import BaseModel, Field, confloat
|
||||
from typing import Optional, List, Dict, Any
|
||||
|
||||
|
||||
class AgentConfig(BaseModel):
|
||||
type: str = Field(..., description="家具类型,如:沙发、餐桌")
|
||||
region: str = Field(..., description="地区/空间,如:客厅、卧室、户外")
|
||||
style: str = Field(..., description="设计风格,如:极简、工业风、中式")
|
||||
temperature: confloat(ge=0, le=2.0) = Field(default=0.7, description="模型温度")
|
||||
|
||||
|
||||
class DeepAgentChatRequest(BaseModel):
|
||||
message: str = Field(..., description="用户的输入指令")
|
||||
thread_id: Optional[str] = Field(None, description="会话线程ID,不传则开启新会话")
|
||||
checkpoint_id: Optional[str] = Field(None, description="回溯点的ID,用于从历史点开启新对话")
|
||||
config_params: Optional[AgentConfig] = None
|
||||
need_suggestion: float = 0
|
||||
need_title: bool = False
|
||||
use_report: bool = False # ← 新增:是否使用深度报告
|
||||
|
||||
|
||||
class HistoryItem(BaseModel):
|
||||
checkpoint_id: str
|
||||
last_message: str
|
||||
node: Optional[str]
|
||||
timestamp: Any
|
||||
|
||||
|
||||
class HistoryResponse(BaseModel):
|
||||
thread_id: str
|
||||
history: List[HistoryItem]
|
||||
|
||||
|
||||
class StreamChunk(BaseModel):
|
||||
node: str
|
||||
content: str
|
||||
checkpoint_id: str
|
||||
Reference in New Issue
Block a user