35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ApiImageGenerateRequest(BaseModel):
|
|
"""图片生成请求。支持全量 Volcano Ark SDK 参数。"""
|
|
|
|
model: str = Field(..., description="模型名称, 如 doubao-seedream-5-0-260128")
|
|
prompt: str = Field(..., description="图片描述提示词")
|
|
size: str | None = Field("2K", description="图片尺寸: 2K | 4K 或 2048x2048")
|
|
response_format: str | None = Field("url", description="返回格式: url | b64_json")
|
|
watermark: bool | None = Field(False, description="是否添加水印")
|
|
image: list[str] | None = Field(None, description="参考图片URL列表")
|
|
output_format: str | None = Field(None, description="输出格式: jpeg | png | webp")
|
|
sequential_image_generation: str | None = Field(
|
|
None, description="组图模式: auto 开启"
|
|
)
|
|
generation_count: int | None = Field(1, ge=1, le=5, description="生成数量: 1-5")
|
|
|
|
|
|
class ApiImageGenerateDataItem(BaseModel):
|
|
"""单张图片结果。"""
|
|
|
|
url: str | None = None
|
|
b64_json: str | None = None
|
|
size: str | None = None
|
|
output_format: str | None = None
|
|
|
|
|
|
class ApiImageGenerateResponse(BaseModel):
|
|
"""图片生成响应（同步返回）。"""
|
|
|
|
created: int
|
|
data: list[ApiImageGenerateDataItem]
|
|
model: str
|