FauziIsyrinApridal commited on
Commit
6968fd4
·
1 Parent(s): 2ec6555
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def create_conversational_chain(vector_store):
89
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True, output_key="answer")
90
  chain = ConversationalRetrievalChain.from_llm(
91
  llm,
92
- retriever=vector_store.as_retriever(search_kwargs={"k": 10}),
93
  combine_docs_chain_kwargs={"prompt": sahabat_prompt},
94
  return_source_documents=True,
95
  memory=memory,
 
89
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True, output_key="answer")
90
  chain = ConversationalRetrievalChain.from_llm(
91
  llm,
92
+ retriever=vector_store.as_retriever(search_kwargs={"k": 6}),
93
  combine_docs_chain_kwargs={"prompt": sahabat_prompt},
94
  return_source_documents=True,
95
  memory=memory,