anhkhoiphan commited on
Commit
80b2a0b
verified
1 Parent(s): 9011670

Update chatbot/core.py

Browse files
Files changed (1) hide show
  1. chatbot/core.py +1 -1
chatbot/core.py CHANGED
@@ -6,7 +6,7 @@ from langchain.chains import ConversationalRetrievalChain
6
 
7
  vector_db = get_vector_db()
8
 
9
- retriever = vector_db.as_retriever(search_kwargs={"k": 15})
10
 
11
  qa_chain = ConversationalRetrievalChain.from_llm(
12
  llm=gemini_llm,
 
6
 
7
  vector_db = get_vector_db()
8
 
9
+ retriever = vector_db.as_retriever(search_kwargs={"k": 20})
10
 
11
  qa_chain = ConversationalRetrievalChain.from_llm(
12
  llm=gemini_llm,