no cache
Browse files
App/Embedding/utils/Initialize.py
CHANGED
|
@@ -57,7 +57,6 @@ def generate_text(doc):
|
|
| 57 |
return pprint.pformat(doc["movie_results"][0]), doc["movie_results"][0]
|
| 58 |
|
| 59 |
|
| 60 |
-
@lru_cache(maxsize=10000)
|
| 61 |
def IdSearch(query: str, background_task: BackgroundTasks):
|
| 62 |
doc = requests.get(
|
| 63 |
f"https://api.themoviedb.org/3/find/{query}?external_source=imdb_id&language=en&api_key={TMDB_API}"
|
|
@@ -71,7 +70,6 @@ def IdSearch(query: str, background_task: BackgroundTasks):
|
|
| 71 |
return TextSearch(text, filter={"key": {"$ne": query}})
|
| 72 |
|
| 73 |
|
| 74 |
-
@lru_cache(maxsize=10000)
|
| 75 |
def TextSearch(query: str, filter=None):
|
| 76 |
docs = docsearch.similarity_search(query, k=10, filter=filter)
|
| 77 |
keys = [doc.metadata["key"] for doc in docs]
|
|
|
|
| 57 |
return pprint.pformat(doc["movie_results"][0]), doc["movie_results"][0]
|
| 58 |
|
| 59 |
|
|
|
|
| 60 |
def IdSearch(query: str, background_task: BackgroundTasks):
|
| 61 |
doc = requests.get(
|
| 62 |
f"https://api.themoviedb.org/3/find/{query}?external_source=imdb_id&language=en&api_key={TMDB_API}"
|
|
|
|
| 70 |
return TextSearch(text, filter={"key": {"$ne": query}})
|
| 71 |
|
| 72 |
|
|
|
|
| 73 |
def TextSearch(query: str, filter=None):
|
| 74 |
docs = docsearch.similarity_search(query, k=10, filter=filter)
|
| 75 |
keys = [doc.metadata["key"] for doc in docs]
|