Quellcode durchsuchen

fix: follow-up (suggested questions) does not refer to the most recen… (#9122)

Hash Brown vor 6 Monaten
Ursprung
Commit
511ffa4698
1 geänderte Dateien mit 5 neuen und 1 gelöschten Zeilen
  1. 5 1
      api/core/memory/token_buffer_memory.py

+ 5 - 1
api/core/memory/token_buffer_memory.py

@@ -58,7 +58,11 @@ class TokenBufferMemory:
         # instead of all messages from the conversation, we only need to extract messages
         # that belong to the thread of last message
         thread_messages = extract_thread_messages(messages)
-        thread_messages.pop(0)
+
+        # 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()
+
         messages = list(reversed(thread_messages))
 
         message_file_parser = MessageFileParser(tenant_id=app_record.tenant_id, app_id=app_record.id)