20 lines
592 B
Python
20 lines
592 B
Python
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ApiModelInfo(BaseModel):
|
|
"""可用模型信息。"""
|
|
|
|
model: str = Field(..., description="模型名称")
|
|
engine_type: str = Field(..., description="引擎类型: video | image")
|
|
engine_id: str = Field(..., description="引擎ID")
|
|
supported_ratios: list[str] | None = None
|
|
supported_resolutions: list[str] | None = None
|
|
supported_durations: list[int] | None = None
|
|
supported_sizes: list[str] | None = None
|
|
|
|
|
|
class ApiModelsResponse(BaseModel):
|
|
"""可用模型列表响应。"""
|
|
|
|
models: list[ApiModelInfo]
|