bug fix
Browse files
App/Embedding/utils/Initialize.py
CHANGED
|
@@ -26,9 +26,7 @@ docsearch = Pinecone.from_existing_index(index_name, embeddings)
|
|
| 26 |
|
| 27 |
|
| 28 |
def check_if_exists(text, imdb_id):
|
| 29 |
-
results = docsearch.similarity_search(
|
| 30 |
-
text, filter={"key": {"$eq": imdb_id}}, top_k=1
|
| 31 |
-
)
|
| 32 |
if results:
|
| 33 |
return True
|
| 34 |
else:
|
|
|
|
| 26 |
|
| 27 |
|
| 28 |
def check_if_exists(text, imdb_id):
|
| 29 |
+
results = docsearch.similarity_search(text, filter={"key": {"$eq": imdb_id}}, k=1)
|
|
|
|
|
|
|
| 30 |
if results:
|
| 31 |
return True
|
| 32 |
else:
|