Update the eviction policy for chat history.

aws-samples · Feb 13, 2025 · b4f5ad9 · b4f5ad9
1 parent b1533a5
commit b4f5ad9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/lambda/online/lambda_main/main_utils/online_entries/common_entry.py b/source/lambda/online/lambda_main/main_utils/online_entries/common_entry.py
@@ -678,7 +678,7 @@ def common_entry(event_body):
     use_history = chatbot_config["use_history"]
     max_rounds_in_memory = event_body["chatbot_config"]["max_rounds_in_memory"]
     # if(len(event_body["chat_history"])<=2*max_rounds_in_memory)
-    chat_history = event_body["chat_history"] if len(event_body["chat_history"])<=2*max_rounds_in_memory else event_body["chat_history"][-2*max_rounds_in_memory:] if use_history else []
+    chat_history = event_body["chat_history"][-2*max_rounds_in_memory:] if use_history else []
     stream = event_body["stream"]
     message_id = event_body["custom_message_id"]
     ws_connection_id = event_body["ws_connection_id"]