twimbit-ai commited on
Commit
11866a8
·
1 Parent(s): 9aa0485

Update query_data.py

Browse files
Files changed (1) hide show
  1. query_data.py +1 -1
query_data.py CHANGED
@@ -65,6 +65,6 @@ def get_chain(vectorstore):
65
 
66
  qa_chain = ConversationalRetrievalChain(return_source_documents=True,
67
  retriever=vectorstore.as_retriever(
68
- search_kwargs={"k": 5, "include_metadata": True}),
69
  combine_docs_chain=doc_chain, question_generator=question_generator)
70
  return qa_chain
 
65
 
66
  qa_chain = ConversationalRetrievalChain(return_source_documents=True,
67
  retriever=vectorstore.as_retriever(
68
+ search_kwargs={"k": 2, "include_metadata": True}),
69
  combine_docs_chain=doc_chain, question_generator=question_generator)
70
  return qa_chain