Bläddra i källkod

fix: remove the latest message from the user that does not have any answer yet (#9297)

kurokobo 6 månader sedan
förälder
incheckning
857055b797
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      api/core/memory/token_buffer_memory.py

+ 2 - 2
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))