File size: 436 Bytes
eca7e38 4db178c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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")
|