Spaces:
Running
Running
File size: 1,325 Bytes
1005046 c46f3e8 1005046 c46f3e8 1005046 c46f3e8 1005046 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import os
from pathlib import Path
from dotenv import load_dotenv
load_dotenv()
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
QDRANT_ENDPOINT = os.getenv("QDRANT_ENDPOINT")
# QDRANT_MOVIE_COLLECTION_NAME = os.getenv("QDRANT_MOVIE_COLLECTION_NAME_BGE")
# QDRANT_TV_COLLECTION_NAME = os.getenv("QDRANT_TV_COLLECTION_NAME_BGE")
QDRANT_MOVIE_COLLECTION_NAME = "Movies_BGE_June"
QDRANT_TV_COLLECTION_NAME = "TV_Shows_BGE_June"
SUPABASE_URL = os.getenv("SUPABASE_URL")
SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
NLTK_PATH = Path(__file__).resolve().parent.parent.parent / "data" / "nltk_data"
BM25_PATH = Path(__file__).resolve().parent.parent.parent / "data" / "bm25_files"
INTENT_MODEL = "JJTsao/intent-classifier-distilbert-movierec" # Fine-tuned intent classification model for query intent classifiation
EMBEDDING_MODEL = "JJTsao/fine-tuned_movie_retriever-bge-base-en-v1.5" # Fine-tuned sentence transfomer model for query dense vector embedding
OPENAI_MODEL = "gpt-4o-mini" # LLM for chat completions
if not OPENAI_API_KEY or not QDRANT_API_KEY:
raise ValueError("Missing API key(s).")
if (
not QDRANT_ENDPOINT
or not QDRANT_MOVIE_COLLECTION_NAME
or not QDRANT_TV_COLLECTION_NAME
):
raise ValueError("Missing QDrant URL or collection name.")
|