from os import getenv
from dotenv import load_dotenv

from pydantic import BaseSettings


class Config(BaseSettings):
    """Configuration for the application."""
    # Pinecone
    pinecone_api_key: str
    pinecone_env: str
    pinecone_index: str
    # LLM
    embedding_dim: int
    embedding_version: str
    embedding_dir: str
    model_name: str
    # HF
    password: str


load_dotenv()
config = Config(
    pinecone_api_key=getenv("PINECONE_API_KEY"),
    pinecone_env=getenv("PINECONE_ENV"),
    pinecone_index=getenv("PINECONE_INDEX"),
    embedding_dim=int(getenv("EMBEDDING_DIM")),
    embedding_version=getenv("EMBEDDING_VERSION"),
    embedding_dir=getenv("EMBEDDING_DIR"),
    model_name=getenv("MODEL_NAME"),
    password=getenv("PASSWORD"),
)