From 857055b797b88fd0fc01a4f8423acd3c399454d5 Mon Sep 17 00:00:00 2001 From: kurokobo Date: Mon, 14 Oct 2024 00:25:50 +0900 Subject: [PATCH] fix: remove the latest message from the user that does not have any answer yet (#9297) --- api/core/memory/token_buffer_memory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/memory/token_buffer_memory.py b/api/core/memory/token_buffer_memory.py index 72da3b0c6f..bc94912c1e 100644 --- a/api/core/memory/token_buffer_memory.py +++ b/api/core/memory/token_buffer_memory.py @@ -60,8 +60,8 @@ class TokenBufferMemory: thread_messages = extract_thread_messages(messages) # for newly created message, its answer is temporarily empty, we don't need to add it to memory - if thread_messages and not thread_messages[-1].answer: - thread_messages.pop() + if thread_messages and not thread_messages[0].answer: + thread_messages.pop(0) messages = list(reversed(thread_messages))