querymind-api / backend /config.py
Usman Bari
Initial commit
37a6ee1
Raw
History Blame Contribute Delete
1.03 kB
import os
from dotenv import load_dotenv
# Base directory is the root of the project (parent of 'backend')
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# Load .env file from base directory
env_path = os.path.join(BASE_DIR, ".env")
load_dotenv(env_path)
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
GROQ_MODEL = "llama-3.1-8b-instant"
DB_DIR = os.path.join(BASE_DIR, "databases", "csv")
SCHEMA_DB_DIR = os.path.join(BASE_DIR, "databases", "schema")
SAMPLE_DATA_DIR = os.path.join(BASE_DIR, "sample_data")
MAX_ROWS_RETURNED = 500
MAX_SQL_RETRIES = 3
# Auto-create directories if missing
os.makedirs(DB_DIR, exist_ok=True)
os.makedirs(SCHEMA_DB_DIR, exist_ok=True)
os.makedirs(SAMPLE_DATA_DIR, exist_ok=True)
print(f"[CONFIG] Base Directory: {BASE_DIR}")
print(f"[CONFIG] CSV Database Directory: {DB_DIR}")
print(f"[CONFIG] Schema Database Directory: {SCHEMA_DB_DIR}")
print(f"[CONFIG] Sample Data Directory: {SAMPLE_DATA_DIR}")
print(f"[CONFIG] Groq API Key loaded: {'Yes' if GROQ_API_KEY else 'No'}")