from pydantic import BaseModel


class ModelDownloadTaskResponse(BaseModel):
    task_id: str
    model_id: str
    status: str
    use_modelscope: bool = False
    path: str | None = None
    error: str | None = None
    progress: float = 0.0
    created_at: str = ""


class DatasetDownloadTaskResponse(BaseModel):
    task_id: str
    dataset_id: str
    status: str
    use_modelscope: bool = False
    path: str | None = None
    error: str | None = None
    record_count: int = 0
    created_at: str = ""