from enum import StrEnum

from pydantic_settings import BaseSettings, SettingsConfigDict


class Model(StrEnum):
    """Model to use for the LLM Chat application."""

    GPT3 = "gpt-3.5-turbo"
    GPT4 = "gpt-4"


class OpenAISettings(BaseSettings):
    """Settings for the LLM Chat application."""

    api_key: str
    model: Model = Model.GPT3
    temperature: float = 0.7

    model_config: SettingsConfigDict = SettingsConfigDict(
        env_file=".env",
        env_file_encoding="utf-8",
        env_prefix="OPENAI_",
        frozen=True,
        use_enum_values=True,
    )