Spaces:
Runtime error
Runtime error
File size: 1,030 Bytes
37a6ee1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | 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'}")
|