Course_rec / app /config.py
Tao Wu
initial
3430cbb
raw
history blame
No virus
731 Bytes
import os
# Redis configuration
REDIS_HOST = os.getenv('REDIS_HOST', 'redis')
REDIS_PORT = int(os.getenv('REDIS_PORT', 6379))
REDIS_DB = int(os.getenv('REDIS_DB', 0))
# Model and embedding configuration
MODEL_NAME = os.getenv('MODEL_NAME', "intfloat/multilingual-e5-large-instruct")
ENCODE_KWARGS = {
'normalize_embeddings': os.getenv('NORMALIZE_EMBEDDINGS', 'True') == 'True',
'convert_to_tensor': os.getenv('CONVERT_TO_TENSOR', 'True') == 'True'
}
QUERY_INSTRUCTION = os.getenv('QUERY_INSTRUCTION', '')
# Other configurations
TOP_K = int(os.getenv('TOP_K', 5))
PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/course_emb_db")
CSV_FILE_PATH = os.getenv('CSV_FILE_PATH', '/app/data/occupations_de.csv')