feat: 新增flux2klein作为moodboard的localbase 模型 ; fix:

app/schemas/generate_image.py

@@ -1,6 +1,6 @@
-from typing import List
+from typing import List, Optional
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 
 class GenerateMultiViewModel(BaseModel):
@@ -8,8 +8,20 @@ class GenerateMultiViewModel(BaseModel):
     image_url: str
 
 
+class GenerateImageFlux2KleinModel(BaseModel):
+    bucket_name: str = Field(..., description="OSS桶名，不传则为None")
+    object_name: str = Field(..., description="OSS对象名（文件路径），不传则为None")
+    # input_image_paths: Optional[List[str]] = Field(default=[], description="输入图片路径列表")
+    width: Optional[int] = Field(default=1024, description="图片宽度，默认512像素")
+    height: Optional[int] = Field(default=1024, description="图片高度，默认512像素")
+    prompt: Optional[str] = Field(default="", description="文本提示词，用于模型推理等场景")
+    steps: Optional[int] = Field(default=4, description="推理步数，控制模型生成过程的迭代次数")
+    guidance: Optional[float] = Field(default=4.0, description="引导系数，调节提示词对生成结果的影响程度")
+
+
 class GenerateImageModel(BaseModel):
-    tasks_id: str
+    bucket_name: str
+    object_name: str
     prompt: str
     image_url: str
     mode: str