From 7293814254d7ffad7c94271d9ee3d3572c72db26 Mon Sep 17 00:00:00 2001 From: JisanAR Date: Sat, 13 Jul 2024 00:25:08 +0600 Subject: [PATCH] retriever condition update --- website/bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/website/bot.py b/website/bot.py index 5957b730f..a58c3cab4 100644 --- a/website/bot.py +++ b/website/bot.py @@ -162,9 +162,9 @@ def conversation_chain(vector_store): ) ) llm = ChatOpenAI(model_name="gpt-3.5-turbo-0125", temperature=0.5) - retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 3}) + retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 5}) memory = ConversationSummaryMemory( - llm=llm, return_messages=True, memory_key="chat_history", max_token_limit=1000 + llm=llm, return_messages=True, memory_key="chat_history", max_token_limit=1500 ) crc = ConversationalRetrievalChain.from_llm(