from pydantic import BaseModel


class FinetuneStartRequest(BaseModel):
    jsonl_url: str
    base_model: str
    hyperparams: dict | None = None


class FinetuneStartResponse(BaseModel):
    job_id: str


class FinetuneStatusResponse(BaseModel):
    job_id: str
    status: str
    progress: int | None = None
    error_message: str | None = None