anhkhoiphan commited on
Commit
94fcb0e
verified
1 Parent(s): f09971c

Update chatbot/core.py

Browse files
Files changed (1) hide show
  1. chatbot/core.py +1 -1
chatbot/core.py CHANGED
@@ -9,7 +9,7 @@ retriever = WikipediaRetriever()
9
 
10
  qa_chain = ConversationalRetrievalChain.from_llm(
11
  llm=gemini_llm,
12
- retriever=get_retriever, # Dynamic Wikipedia search
13
  memory=memory,
14
  return_source_documents=False,
15
  combine_docs_chain_kwargs={"prompt": chat_prompt},
 
9
 
10
  qa_chain = ConversationalRetrievalChain.from_llm(
11
  llm=gemini_llm,
12
+ retriever=retriever, # Dynamic Wikipedia search
13
  memory=memory,
14
  return_source_documents=False,
15
  combine_docs_chain_kwargs={"prompt": chat_prompt},