File size: 532 Bytes
92e030f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# config.py
# This file stores all the configuration variables and constants for the application.

# Embedding Model Configuration
MODEL_NAME = "jinaai/jina-embeddings-v3"
MODEL_KWARGS = {'device': 'cpu', 'trust_remote_code': True}
ENCODE_KWARGS = {'normalize_embeddings': False}

# Vector Store Configuration
VECTOR_STORE_DIRECTORY = "vectorStore"

# LLM Configuration
LLM_MODEL = "llama-3.3-70b-versatile"
LLM_TEMPERATURE = 0.8

# Retriever Configuration
DENSE_RETRIEVER_K = 3
KEYWORD_RETRIEVER_K = 3
ENSEMBLE_WEIGHTS = [0.5, 0.5]