from pydantic import BaseModel


class ProviderAccountPayload(BaseModel):
    provider_code: str
    provider_name: str
    api_format: str
    base_url: str
    api_key: str
    api_secret: str | None = ""
    webhook_secret: str | None = ""
    timeout_seconds: int = 120
    max_retries: int = 3
    status: int = 1
    remark: str = ""


class ProviderModelPayload(BaseModel):
    provider_account_id: int
    model_code: str
    model_name: str
    request_content_type: str = "application/json"
    supports_text_to_video: bool = True
    supports_image_to_video: bool = False
    supports_video_reference: bool = False
    supports_audio_reference: bool = False
    supports_generate_audio: bool = False
    supports_remix: bool = False
    supports_webhook: bool = False
    min_duration: int = 4
    max_duration: int = 12
    default_ratio: str = "16:9"
    default_resolution: str = "1280x720"
    status: int = 1