feat 1.增加 建议词 机制 2.对话生图实现
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user