File size: 517 Bytes
05d3b70
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import os

# --- Directory Paths ---
DATA_DIR = os.path.join("data")
RAW_DIR = os.path.join(DATA_DIR, "raw")
INDEX_DIR = os.path.join(DATA_DIR, "indices")
INDEX_PATH = os.path.join(INDEX_DIR, "mindmesh.index")
META_PATH = os.path.join(INDEX_DIR, "meta.json")

# --- Model Names ---
EMBEDDING_MODEL = "all-MiniLM-L6-v2"
SUMMARIZER_MODEL = "facebook/bart-large-cnn"
PRIMARY_GROQ_MODEL = "llama-3.1-70b-versatile"
FALLBACK_GROQ_MODEL = "llama-3.1-8b-instant"

# --- Other Settings ---
MAX_TOKENS = 6000
TOP_K_RESULTS = 5