Upload extractor.py

extractor.py

@@ -43,7 +43,7 @@ def _prompt(question):
             Context: """
 
 
-def _search(query, extractor, question=None):
+async def _search(query, extractor, question=None):
     # Default question to query if empty
     if not question:
         question = query
@@ -89,6 +89,6 @@ async def rag(question: str):
 
     # Create extractor instance
     extractor = Extractor(embeddings, "google/flan-t5-base")
-    answer = _search(question, extractor)
+    answer = await _search(question, extractor)
     # print(question, answer)
     return {answer}