tlong-ds's picture
update
c793d00
raw
history blame contribute delete
705 Bytes
import google.generativeai as genai
from langchain.llms.base import LLM
from typing import List, Optional
from dotenv import load_dotenv
import os
load_dotenv()
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
QDRANT_HOST = os.getenv("QDRANT_HOST")
QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
QDRANT_COLLECTION_NAME = "dbms_collection_courses"
QDRANT_COLLECTION_NAME_LECTURES = "dbms_collection_lectures"
MODEL_NAME = "gemini-2.0-flash"
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
EMBEDDING_SIZE = 384
HF_HOME = "/tmp/huggingface"
TRANSFORMERS_CACHE = "/tmp/huggingface"
TORCH_HOME = "/tmp/torch"
SENTENCE_TRANSFORMERS_HOME = "/tmp/sentence_transformers"
MPLCONFIGDIR = "/tmp/matplotlib"