config/config.py

import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Configuration class
class Config:
    API_KEY = os.getenv('API_KEY')
    OPENAI_API_BASE = os.getenv("OPENAI_API_BASE") 
    GROQ_API_KEY = os.getenv('GROQ_API_KEY')
    MEM0_API_KEY = os.getenv('mem0')
    
    # Model configurations
    EMBEDDING_MODEL = "text-embedding-ada-002"
    CHAT_MODEL = "gpt-4o-mini"
    LLAMA_GUARD_MODEL = "meta-llama/llama-guard-4-12b"
    
    # Vector store configuration
    COLLECTION_NAME = "nutritional_hypotheticals"
    PERSIST_DIRECTORY = "./data/nutritional_db"
    
    # Agent configuration
    GROUNDEDNESS_THRESHOLD = 0.4
    PRECISION_THRESHOLD = 0.7
    MAX_ITERATIONS = 2
    RETRIEVAL_K = 3

config = Config()