Zubaish commited on
Commit
d322d09
·
1 Parent(s): cd319c6

Fix: define KB_DIR in config

Browse files
Files changed (1) hide show
  1. config.py +24 -4
config.py CHANGED
@@ -1,5 +1,25 @@
1
- # config.py
2
 
3
- KB_DIR = "kb"
4
- VECTOR_DIR = "vectorstore"
5
- EMBED_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
 
3
+ # -----------------------
4
+ # Paths
5
+ # -----------------------
6
+ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
7
+
8
+ # Your knowledge base folder (this MUST exist in the repo)
9
+ KB_DIR = os.path.join(BASE_DIR, "kb")
10
+
11
+ # Chroma persistence directory
12
+ CHROMA_DIR = os.path.join(BASE_DIR, "chroma_db")
13
+
14
+ # -----------------------
15
+ # Models
16
+ # -----------------------
17
+ EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
18
+ LLM_MODEL = "microsoft/Phi-3-mini-4k-instruct"
19
+
20
+ # -----------------------
21
+ # RAG params
22
+ # -----------------------
23
+ CHUNK_SIZE = 500
24
+ CHUNK_OVERLAP = 50
25
+ TOP_K = 3