raviix46 commited on
Commit
05d3b70
·
verified ·
1 Parent(s): 33b6cbb

Create config.py

Browse files
Files changed (1) hide show
  1. src/config.py +18 -0
src/config.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ # --- Directory Paths ---
4
+ DATA_DIR = os.path.join("data")
5
+ RAW_DIR = os.path.join(DATA_DIR, "raw")
6
+ INDEX_DIR = os.path.join(DATA_DIR, "indices")
7
+ INDEX_PATH = os.path.join(INDEX_DIR, "mindmesh.index")
8
+ META_PATH = os.path.join(INDEX_DIR, "meta.json")
9
+
10
+ # --- Model Names ---
11
+ EMBEDDING_MODEL = "all-MiniLM-L6-v2"
12
+ SUMMARIZER_MODEL = "facebook/bart-large-cnn"
13
+ PRIMARY_GROQ_MODEL = "llama-3.1-70b-versatile"
14
+ FALLBACK_GROQ_MODEL = "llama-3.1-8b-instant"
15
+
16
+ # --- Other Settings ---
17
+ MAX_TOKENS = 6000
18
+ TOP_K_RESULTS = 5