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"