import os
from dotenv import load_dotenv

load_dotenv()

def get_config():
    return {
        "MONGO_URI": os.getenv("MONGO_URI"),
        "DB_NAME": os.getenv("DB_NAME"),
        "COLLECTION_USERS": os.getenv("COLLECTION_USERS"),
        "COLLECTION_CHAT": os.getenv("COLLECTION_CHAT"),
        "OLLAMA_MODEL": os.getenv("OLLAMA_MODEL"),
        "OLLAMA_KEEPALIVE": os.getenv("OLLAMA_KEEPALIVE"),
        "MAX_HISTORY_LIMIT": int(os.getenv("MAX_HISTORY_LIMIT", "50")),
        "SYSTEM_PROMPT": os.getenv("SYSTEM_PROMPT"),
    }