retriever condition update

OWASP-BLT · Jul 12, 2024 · 7293814 · 7293814
1 parent 3a62413
commit 7293814
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(