from datetime import datetime
from typing import Optional

from pydantic import BaseModel


class Sample(BaseModel):
    filename: str
    url: str
    created_at: datetime
    step: Optional[int] = None