from datetime import datetime
from typing import Optional

from pydantic import BaseModel


class TrainingStatus(BaseModel):
    current_step: int
    total_steps: int
    loss: float
    learning_rate: float
    percentage: float
    eta_seconds: Optional[float]
    steps_per_second: float
    updated_at: datetime
    source: str  # 'local' or 'remote'
    source_path: str