import enum

from pydantic_settings import BaseSettings


class LogLevel(str, enum.Enum):
    """Possible log levels."""

    NOTSET = "NOTSET"
    DEBUG = "DEBUG"
    INFO = "INFO"
    WARNING = "WARNING"
    ERROR = "ERROR"
    FATAL = "FATAL"


class Settings(BaseSettings):
    temperature: float = 1
    model: str = "llama3-8b-8192"
    log_level: LogLevel = LogLevel.INFO
    segment_time: int = 1500


app_settings = Settings()