feat 1.增加 建议词 机制 2.对话生图实现

This commit is contained in:
zcr
2026-02-06 11:55:11 +08:00
parent ec195d17e1
commit 3248c45cd4
12 changed files with 655 additions and 85 deletions

View File

@@ -1,11 +1,20 @@
from pydantic import BaseModel, Field
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 ChatRequest(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
# force_sketch: bool = False # 新增:是否强制绘图
class HistoryItem(BaseModel):