JJTsao commited on
Commit
ff71b86
·
verified ·
1 Parent(s): ab11aba

Update app/core/config.py

Browse files
Files changed (1) hide show
  1. app/core/config.py +3 -5
app/core/config.py CHANGED
@@ -9,10 +9,8 @@ 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
- 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")
@@ -25,7 +23,7 @@ EMBEDDING_MODEL = "JJTsao/fine-tuned_movie_retriever-bge-base-en-v1.5" # Fine-t
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
 
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")
 
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