feat(新功能): batch generate relight 入参回参修改

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试):
This commit is contained in:
zchengrong
2025-06-05 15:14:36 +08:00
parent 90f9879edb
commit 12bb128351
2 changed files with 96 additions and 61 deletions

View File

@@ -45,6 +45,7 @@ class GenerateRelightImageModel(BaseModel):
"""
# product任务子项
class ProductItemModel(BaseModel):
tasks_id: str
image_strength: float
@@ -53,16 +54,24 @@ class ProductItemModel(BaseModel):
product_type: str
# product批处理 集合
class BatchGenerateProductImageModel(BaseModel):
batch_tasks_id: str
user_id: str
batch_data_list: List[ProductItemModel]
class BatchGenerateRelightImageModel(BaseModel):
# relight任务子项
class RelightItemModel(BaseModel):
tasks_id: str
prompt: str
image_url: str
direction: str
product_type: str
batch_size: int
# relight批处理集合
class BatchGenerateRelightImageModel(BaseModel):
batch_tasks_id: str
user_id: str
batch_data_list: List[RelightItemModel]