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="用户的输入指令")
    # image_url: Optional[str] = Field(None, 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
    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