from typing import Optional

from pydantic import BaseModel, Field


class FashionAgentRequest(BaseModel):
    """服装设计 Agent 请求"""

    message: str = Field(default="", description="用户输入的消息")
    user_id: str = Field(default="test-agent", description="用户ID,用于生成图片存储路径")
    enable_thinking: bool = Field(default=False, description="模型思考是否开启")

    call_print: bool = Field(default=False, description="是否直接调用 print 生成印花")
    print_need_prompt_generation: bool = Field(default=False, description="print 是否需要 LLM 生成 prompt")

    call_logo: bool = Field(default=False, description="是否直接调用 logo 生成装饰图案")

    call_sketch: bool = Field(default=False, description="是否直接调用 sketch 生成草图")
    sketch_need_prompt_generation: bool = Field(default=False, description="sketch 是否需要 LLM 生成 prompt")

    call_design: bool = Field(default=False, description="是否直接调用 design 生成设计系列")
    design_request_data: dict = Field(default={}, description="design 请求参数")

    call_trending: bool = Field(default=False, description="是否直接调用 trending 趋势分析")
    call_explor: bool = Field(default=False, description="是否直接调用 explorer 灵感探索")

    provider: Optional[str] = Field(default="unsplash", description="图片源: pexels 或 unsplash")