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]