zcr
b9163f0b46
- 更新了“generate_image.py”以接受输入图像以增强图像生成。 - 修改了`pexels_search.py`和`unsplash_search.py`以将日志记录和上传路径从“explorer”更改为“explore”。 - 调整了“print_graph”和“sketch_graph”以提取最新的用户输入并处理输入图像以生成打印和草图图像。 - 重构“generate_print_tool”和“generate_sketch_tool”以接受输入图像。 - 更新了“main_agent.py”以包含状态中的输入图像并调整了图形构建过程。 - 增强了“service.py”来管理输入图像并改进了流媒体期间的事件处理。 - 更新了新软件包和版本的“pyproject.toml”和“uv.lock”中的依赖项。
31 lines
1.5 KiB
Python
31 lines
1.5 KiB
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class FashionAgentRequest(BaseModel):
|
|
"""服装设计 Agent 请求"""
|
|
|
|
thread_id: str = Field(description="会话id")
|
|
|
|
message: str = Field(default="", description="用户输入的消息")
|
|
user_id: str = Field(default="test-agent", description="用户ID,用于生成图片存储路径")
|
|
input_images: list[str] = Field(default=[], description="用户上传图片地址")
|
|
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_explore: bool = Field(default=False, description="是否直接调用 explore 灵感探索")
|
|
|
|
provider: Optional[str] = Field(default="unsplash", description="图片源: pexels 或 unsplash")
|