from typing import Optional
from pydantic import BaseModel


class OllamaModelStatus(BaseModel):
    name: str
    parameters: Optional[str] = None
    size: Optional[int] = None
    downloaded: Optional[int] = None
    status: str
    done: bool
    error: Optional[str] = None