Dinesh310 commited on
Commit
4432a1c
·
verified ·
1 Parent(s): 7767ecf

Create config/config.py

Browse files
Files changed (1) hide show
  1. src/config/config.py +21 -0
src/config/config.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ class Config:
4
+ # -------- Embeddings --------
5
+ EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
6
+ EMBEDDING_DEVICE = "cpu"
7
+ NORMALIZE_EMBEDDINGS = True
8
+
9
+ # -------- LLM --------
10
+ LLM_MODEL = "openai/gpt-oss-120b:free"
11
+ OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1"
12
+ OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
13
+
14
+ # -------- Text Splitter --------
15
+ CHUNK_SIZE = 500
16
+ CHUNK_OVERLAP = 100
17
+
18
+ # -------- Retriever --------
19
+ SEARCH_TYPE = "mmr"
20
+ TOP_K = 5
21
+ LAMBDA_MULT = 0.25