twimbit-ai commited on
Commit
a946efd
·
1 Parent(s): 0618345

Update query_data.py

Browse files
Files changed (1) hide show
  1. query_data.py +1 -1
query_data.py CHANGED
@@ -23,7 +23,7 @@ def get_chain(vectorstore):
23
  llm = ChatOpenAI(temperature=0)
24
  qa_chain = ConversationalRetrievalChain.from_llm(
25
  llm,
26
- vectorstore.as_retriever(search_kwargs={"k": 10})
27
  # qa_prompt=QA_PROMPT,
28
  # condense_question_prompt=CONDENSE_QUESTION_PROMPT,
29
  )
 
23
  llm = ChatOpenAI(temperature=0)
24
  qa_chain = ConversationalRetrievalChain.from_llm(
25
  llm,
26
+ vectorstore.as_retriever(search_kwargs={"k": 8, "include_metadata": True})
27
  # qa_prompt=QA_PROMPT,
28
  # condense_question_prompt=CONDENSE_QUESTION_PROMPT,
29
  )