Mbonea commited on
Commit
85075e1
·
1 Parent(s): 7caca4f

added lru cache

Browse files
Files changed (1) hide show
  1. App/Embedding/utils/Initialize.py +3 -0
App/Embedding/utils/Initialize.py CHANGED
@@ -2,6 +2,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
2
  from langchain.docstore.document import Document
3
  from langchain.vectorstores import Pinecone
4
  from fastapi import BackgroundTasks
 
5
  import os, requests
6
  import pinecone, pprint
7
  from .Elastic import FetchDocuments
@@ -56,6 +57,7 @@ def generate_text(doc):
56
  return pprint.pformat(doc["movie_results"][0]), doc["movie_results"][0]
57
 
58
 
 
59
  def IdSearch(query: str, background_task: BackgroundTasks):
60
  doc = requests.get(
61
  f"https://api.themoviedb.org/3/find/{query}?external_source=imdb_id&language=en&api_key={TMDB_API}"
@@ -69,6 +71,7 @@ def IdSearch(query: str, background_task: BackgroundTasks):
69
  return TextSearch(text, filter={"key": {"$ne": query}})
70
 
71
 
 
72
  def TextSearch(query: str, filter=None):
73
  docs = docsearch.similarity_search(query, k=10, filter=filter)
74
  keys = [doc.metadata["key"] for doc in docs]
 
2
  from langchain.docstore.document import Document
3
  from langchain.vectorstores import Pinecone
4
  from fastapi import BackgroundTasks
5
+ from functools import lru_cache
6
  import os, requests
7
  import pinecone, pprint
8
  from .Elastic import FetchDocuments
 
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
  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]