core/config.py

from pydantic_settings import BaseSettings
from typing import Optional
import os
from dotenv import load_dotenv

load_dotenv()

class Settings(BaseSettings):
    GROQ_API_KEY: str = os.getenv("GROQ_API_KEY", "")
    CHROMA_DB_PATH: str = "chroma_db"
    MODEL_NAME: str = "gemma2-9b-it"
    TEMPERATURE: float = 0.7
    MAX_TOKENS: int = 2048
    
    class Config:
        env_file = ".env"

settings = Settings()