File size: 652 Bytes
11ba2bd 5582d21 a2efaea 5582d21 a2efaea 11ba2bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | EMBED_MODEL = "BAAI/bge-small-en-v1.5"
RERANK_MODEL = "BAAI/bge-reranker-base"
CHUNK_SIZE = 700 # tokens
CHUNK_OVERLAP = 120
TOP_K_BM25 = 20
TOP_K_DENSE = 20
TOP_K_FUSED = 30
TOP_N_FINAL = 5
RRF_K = 60
DEFAULT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
HF_MODELS = [
"Qwen/Qwen2.5-Coder-32B-Instruct",
"meta-llama/Llama-3.3-70B-Instruct",
"meta-llama/Llama-3.1-8B-Instruct",
"mistralai/Mistral-7B-Instruct-v0.3",
]
DATA_DIR = "data"
CHUNKS_FILE = DATA_DIR + "/chunks.parquet"
FAISS_FILE = DATA_DIR + "/index.faiss"
BM25_FILE = DATA_DIR + "/bm25.pkl"
WIKI_BASE_URL = "https://wiki.freecad.org"
|