新增标题提取

This commit is contained in:
zcr
2026-03-12 15:19:26 +08:00
parent 510a5117ee
commit 25abdfa38a
3 changed files with 86 additions and 78 deletions

View 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