vibbabackend / config.py
prestiva's picture
UPDATE: book3
92e030f
raw
history blame contribute delete
532 Bytes
# 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]