28 lines
1.4 KiB
Python
28 lines
1.4 KiB
Python
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")
|