Update chatbot/core.py

chatbot/core.py

@@ -3,6 +3,7 @@ from chatbot.memory import memory
 from chatbot.prompts import chat_prompt
 from langchain.retrievers import WikipediaRetriever
 from langchain.chains import ConversationalRetrievalChain
+from langchain.schema import BaseRetriever
 
 def translate_to_english(text: str) -> str:
     """Use Gemini LLM to translate text to English."""
@@ -10,7 +11,7 @@ def translate_to_english(text: str) -> str:
     response = gemini_llm.invoke(prompt)  # Invoke Gemini for translation
     return response  # Assuming `gemini_llm.invoke()` returns plain text
 
-class WikipediaTranslationRetriever:
+class WikipediaTranslationRetriever(BaseRetriever):
     """Custom Retriever that translates queries before searching Wikipedia."""
     def __init__(self, retriever, translator):
         self.retriever = retriever
@@ -21,6 +22,11 @@ class WikipediaTranslationRetriever:
         print(f"🔄 Translated Query: {translated_query}")
         return self.retriever.get_relevant_documents(translated_query)
 
+    async def aget_relevant_documents(self, query):
+        # If your environment doesn't need async support, you can simply raise an error.
+        raise NotImplementedError("Asynchronous retrieval is not implemented.")
+
+
 # ✅ Use WikipediaRetriever
 wiki_retriever = WikipediaRetriever()