Dinesh310 commited on
Commit
4c645b4
·
verified ·
1 Parent(s): 3305f51

Update src/config/config.py

Browse files
Files changed (1) hide show
  1. src/config/config.py +15 -16
src/config/config.py CHANGED
@@ -1,21 +1,20 @@
 
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
 
1
+ # src/config.py
2
  import os
3
 
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
+ MMR_LAMBDA = 0.25
20
+ K_OFFSET = 2