Update chatbot/core.py

chatbot/core.py

@@ -1,9 +1,12 @@
-from chatbot.retrieval import get_retriever
+from langchain.retrievers import WikipediaRetriever
 from chatbot.llm import gemini_llm  
 from chatbot.memory import memory  
 from chatbot.prompts import chat_prompt  
 from langchain.chains import ConversationalRetrievalChain
 
+# Create the retriever
+retriever = WikipediaRetriever()
+
 qa_chain = ConversationalRetrievalChain.from_llm(
     llm=gemini_llm, 
     retriever=get_retriever,  # Dynamic Wikipedia search