mayar-waleed commited on
Commit
4c935cf
·
1 Parent(s): 61b04dd

Add dockerfile

Browse files
Files changed (1) hide show
  1. app/config.py +7 -4
app/config.py CHANGED
@@ -8,17 +8,20 @@ from dotenv import load_dotenv
8
 
9
  load_dotenv()
10
 
 
 
 
11
 
12
  @dataclass(frozen=True)
13
  class Settings:
14
- base_dir: str = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
15
- data_dir: str = os.path.join(base_dir, "data")
16
- chroma_dir: str = os.path.join(base_dir, "chroma_db")
17
 
18
  groq_api_key: str = os.getenv("GROQ_API_KEY", "")
19
  groq_model_name: str = os.getenv("GROQ_MODEL_NAME", "llama-3.3-70b-versatile")
20
 
21
- reranker_model_path: str = os.getenv("RERANKER_MODEL_PATH", "")
22
 
23
  semantic_k: int = int(os.getenv("SEMANTIC_K", "10"))
24
  bm25_k: int = int(os.getenv("BM25_K", "10"))
 
8
 
9
  load_dotenv()
10
 
11
+ # Compute base directory
12
+ _BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
13
+
14
 
15
  @dataclass(frozen=True)
16
  class Settings:
17
+ base_dir: str = _BASE_DIR
18
+ data_dir: str = os.path.join(_BASE_DIR, "data")
19
+ chroma_dir: str = os.path.join(_BASE_DIR, "chroma_db")
20
 
21
  groq_api_key: str = os.getenv("GROQ_API_KEY", "")
22
  groq_model_name: str = os.getenv("GROQ_MODEL_NAME", "llama-3.3-70b-versatile")
23
 
24
+ reranker_model_path: str = os.getenv("RERANKER_MODEL_PATH", os.path.join(_BASE_DIR, "reranker"))
25
 
26
  semantic_k: int = int(os.getenv("SEMANTIC_K", "10"))
27
  bm25_k: int = int(os.getenv("BM25_K", "10"))