Update processing.py

processing.py

@@ -1,4 +1,4 @@
-from langchain.schema import HumanMessage, BaseRetriever
+from langchain.schema import HumanMessage, BaseRetriever, Document
 from output_parser import output_parser
 from langchain_openai import OpenAIEmbeddings, ChatOpenAI
 from langchain_community.vectorstores import FAISS
@@ -51,10 +51,20 @@ class CombinedRetriever(BaseRetriever):
     class Config:
         arbitrary_types_allowed = True
 
-    def invoke(self, input):
+    def get_relevant_documents(self, query: str) -> List[Document]:
+        return self.invoke(query)
+
+    async def aget_relevant_documents(self, query: str) -> List[Document]:
+        combined_docs = []
+        for retriever in self.retrievers:
+            docs = await retriever.aget_relevant_documents(query)
+            combined_docs.extend(docs)
+        return combined_docs
+
+    def invoke(self, query: str) -> List[Document]:
         combined_docs = []
         for retriever in self.retrievers:
-            docs = retriever.invoke(input)
+            docs = retriever.get_relevant_documents(query)
             combined_docs.extend(docs)
         return combined_docs