cache failed
Browse files
App/Embedding/utils/Initialize.py
CHANGED
|
@@ -27,7 +27,7 @@ vector_index = pinecone.Index(index_name=index_name)
|
|
| 27 |
docsearch = Pinecone.from_existing_index(index_name, embeddings)
|
| 28 |
|
| 29 |
|
| 30 |
-
@cache(namespace="test")
|
| 31 |
def check_if_exists(text, imdb_id):
|
| 32 |
results = docsearch.similarity_search(text, filter={"key": {"$eq": imdb_id}}, k=1)
|
| 33 |
if results:
|
|
@@ -36,7 +36,7 @@ def check_if_exists(text, imdb_id):
|
|
| 36 |
return False
|
| 37 |
|
| 38 |
|
| 39 |
-
@cache(namespace="test")
|
| 40 |
def add_document(imdb_id, doc):
|
| 41 |
text, temp_doc = doc
|
| 42 |
response = check_if_exists(text=text, imdb_id=imdb_id)
|
|
@@ -54,7 +54,7 @@ def add_document(imdb_id, doc):
|
|
| 54 |
docsearch.add_documents([temp])
|
| 55 |
|
| 56 |
|
| 57 |
-
@cache(namespace="test")
|
| 58 |
def generate_text(doc):
|
| 59 |
if doc["tv_results"]:
|
| 60 |
return pprint.pformat(doc["tv_results"][0]), doc["tv_results"][0]
|
|
|
|
| 27 |
docsearch = Pinecone.from_existing_index(index_name, embeddings)
|
| 28 |
|
| 29 |
|
| 30 |
+
# @cache(namespace="test")
|
| 31 |
def check_if_exists(text, imdb_id):
|
| 32 |
results = docsearch.similarity_search(text, filter={"key": {"$eq": imdb_id}}, k=1)
|
| 33 |
if results:
|
|
|
|
| 36 |
return False
|
| 37 |
|
| 38 |
|
| 39 |
+
# @cache(namespace="test")
|
| 40 |
def add_document(imdb_id, doc):
|
| 41 |
text, temp_doc = doc
|
| 42 |
response = check_if_exists(text=text, imdb_id=imdb_id)
|
|
|
|
| 54 |
docsearch.add_documents([temp])
|
| 55 |
|
| 56 |
|
| 57 |
+
# @cache(namespace="test")
|
| 58 |
def generate_text(doc):
|
| 59 |
if doc["tv_results"]:
|
| 60 |
return pprint.pformat(doc["tv_results"][0]), doc["tv_results"][0]
|