from pydantic import BaseModel, Field


class CreateGenerationTaskRequest(BaseModel):
    workspace_id: str = Field(..., min_length=1)
    project_id: str = Field(..., min_length=1)
    asset_library_id: str = Field(..., min_length=1)
    strategy_id: str = ""
    voice_library_id: str = ""
    edit_plan_id: str = ""
    created_by_user_id: str = ""


class GenerationTaskResponse(BaseModel):
    id: str
    workspace_id: str
    project_id: str
    asset_library_id: str
    strategy_id: str
    voice_library_id: str
    edit_plan_id: str
    status: str
    progress: float
    result_count: int
    error_message: str