JJ Tsao commited on
Commit
c46f3e8
·
1 Parent(s): 1080531

Update config.py

Browse files
Files changed (1) hide show
  1. app/core/config.py +6 -4
app/core/config.py CHANGED
@@ -9,8 +9,10 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
9
  QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
10
 
11
  QDRANT_ENDPOINT = os.getenv("QDRANT_ENDPOINT")
12
- QDRANT_MOVIE_COLLECTION_NAME = os.getenv("QDRANT_MOVIE_COLLECTION_NAME_BGE")
13
- QDRANT_TV_COLLECTION_NAME = os.getenv("QDRANT_TV_COLLECTION_NAME_BGE")
 
 
14
 
15
  SUPABASE_URL = os.getenv("SUPABASE_URL")
16
  SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
@@ -18,12 +20,12 @@ SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
18
  NLTK_PATH = Path(__file__).resolve().parent.parent.parent / "data" / "nltk_data"
19
  BM25_PATH = Path(__file__).resolve().parent.parent.parent / "data" / "bm25_files"
20
 
21
- INTENT_MODEL = "JJTsao/intent-classifier-distilbert-moviebot" # Fine-tuned intent classification model for query intent classifiation
22
  EMBEDDING_MODEL = "JJTsao/fine-tuned_movie_retriever-bge-base-en-v1.5" # Fine-tuned sentence transfomer model for query dense vector embedding
23
  OPENAI_MODEL = "gpt-4o-mini" # LLM for chat completions
24
 
25
 
26
- if not OPENAI_API_KEY or not QDRANT_API_KEY or not SUPABASE_API_KEY:
27
  raise ValueError("Missing API key(s).")
28
  if (
29
  not QDRANT_ENDPOINT
 
9
  QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
10
 
11
  QDRANT_ENDPOINT = os.getenv("QDRANT_ENDPOINT")
12
+ # QDRANT_MOVIE_COLLECTION_NAME = os.getenv("QDRANT_MOVIE_COLLECTION_NAME_BGE")
13
+ # QDRANT_TV_COLLECTION_NAME = os.getenv("QDRANT_TV_COLLECTION_NAME_BGE")
14
+ QDRANT_MOVIE_COLLECTION_NAME = "Movies_BGE_June"
15
+ QDRANT_TV_COLLECTION_NAME = "TV_Shows_BGE_June"
16
 
17
  SUPABASE_URL = os.getenv("SUPABASE_URL")
18
  SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
 
20
  NLTK_PATH = Path(__file__).resolve().parent.parent.parent / "data" / "nltk_data"
21
  BM25_PATH = Path(__file__).resolve().parent.parent.parent / "data" / "bm25_files"
22
 
23
+ INTENT_MODEL = "JJTsao/intent-classifier-distilbert-movierec" # Fine-tuned intent classification model for query intent classifiation
24
  EMBEDDING_MODEL = "JJTsao/fine-tuned_movie_retriever-bge-base-en-v1.5" # Fine-tuned sentence transfomer model for query dense vector embedding
25
  OPENAI_MODEL = "gpt-4o-mini" # LLM for chat completions
26
 
27
 
28
+ if not OPENAI_API_KEY or not QDRANT_API_KEY:
29
  raise ValueError("Missing API key(s).")
30
  if (
31
  not QDRANT_ENDPOINT