Spaces:
Running
Running
JJ Tsao
commited on
Commit
·
c46f3e8
1
Parent(s):
1080531
Update config.py
Browse files- 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-
|
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
|
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
|