44 lines
1.6 KiB
Python
Executable File
44 lines
1.6 KiB
Python
Executable File
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="用户的输入指令")
|
||
enable_thinking: Optional[bool] = Field(default=False, description="是否开启思考模式")
|
||
quote_image_path: Optional[str] = Field(None, description="引用图片地址") # ✅ 新增
|
||
input_image_paths: Optional[list[str]] = Field(None, description="上传图片地址集合") # ✅ 新增
|
||
thread_id: Optional[str] = Field(None, description="会话线程ID，不传则开启新会话")
|
||
checkpoint_id: Optional[str] = Field(None, description="回溯点的ID，用于从历史点开启新对话")
|
||
config_params: Optional[AgentConfig] = Field(
|
||
default=None,
|
||
description="Agent 配置参数（type/region/style 等）"
|
||
)
|
||
need_suggestion: float = 0
|
||
use_report: bool = False # ← 新增：是否使用深度报告
|
||
language: str = "en"
|
||
|
||
|
||
class HistoryItem(BaseModel):
|
||
checkpoint_id: str
|
||
last_message: Any
|
||
node: Optional[str]
|
||
timestamp: Any
|
||
|
||
|
||
class HistoryResponse(BaseModel):
|
||
thread_id: str
|
||
history: List[HistoryItem]
|
||
|
||
|
||
class StreamChunk(BaseModel):
|
||
node: str
|
||
content: str
|
||
checkpoint_id: str
|