File size: 758 Bytes
e23acaf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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()