from typing import List

from pydantic import BaseModel, Field


class SketchToGarmentModel(BaseModel):
    input_image_path: str = Field(..., description="输入图片路径列表")
    bucket_name: str = Field(..., description="输入图片路径列表")
    user_id: str = Field(..., description="用户id")
    callback_url: str  # 必填，客户端提供的回调地址
    task_id: str = Field()
    model: str = Field(default="single", description="模型类型: single 或 multi")