Test / backend /config.py
UKielz's picture
Update backend/config.py
4db178c verified
import os
import torch
from dotenv import load_dotenv
load_dotenv()
COLLECTION_NAME = "cocacola_vietname_data"
EMBEDDING_MODEL_NAME = "AITeamVN/Vietnamese_Embedding"
MAX_LENGTH = 512
TOP_K = 5
MIN_SIMILARITY_SCORE = 0.5
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
QDRANT_URL = os.environ.get("QDRANT_URL")
QDRANT_API_KEY = os.environ.get("QDRANT_API_KEY")
GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")