from typing import Optional
from datetime import datetime
from typing_extensions import Literal

from .._models import BaseModel
from .model_capabilities import ModelCapabilities

__all__ = ["ModelInfo"]


class ModelInfo(BaseModel):
    id: str
    """Unique model identifier."""

    capabilities: Optional[ModelCapabilities] = None
    """Model capability information."""

    created_at: datetime
    """RFC 3339 datetime string representing the time at which the model was released.

    May be set to an epoch value if the release date is unknown.
    """

    display_name: str
    """A human-readable name for the model."""

    max_input_tokens: Optional[int] = None
    """Maximum input context window size in tokens for this model."""

    max_tokens: Optional[int] = None
    """Maximum value for the `max_tokens` parameter when using this model."""

    type: Literal["model"]
    """Object type.

    For Models, this is always `"model"`.
    """