Mind-Mesh / src /config.py
raviix46's picture
Create config.py
05d3b70 verified
raw
history blame contribute delete
517 Bytes
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