ThesisBackend / src /config /settings.py
AdarshRajDS
Fix HF persistent storage paths
e23acaf
from dataclasses import dataclass
from dotenv import load_dotenv
import os
load_dotenv()
# 🔥 This becomes /data on Hugging Face, and stays local when developing
BASE_DATA_DIR = os.getenv("HF_HOME", "data")
@dataclass
class Settings:
app_env: str = os.getenv("APP_ENV", "development")
log_level: str = os.getenv("LOG_LEVEL", "INFO")
# 📂 Data paths
base_data_dir: str = BASE_DATA_DIR
raw_data_dir: str = os.path.join(BASE_DATA_DIR, "raw")
processed_data_dir: str = os.path.join(BASE_DATA_DIR, "processed")
chroma_dir: str = os.path.join(BASE_DATA_DIR, "chroma")
# 🤖 Models
embedding_model: str = "sentence-transformers/all-MiniLM-L6-v2"
llm_model: str = "llama-3.1-8b-instant"
settings = Settings()