浏览代码

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

tags/0.9.2
Hash Brown 1年前
父节点
当前提交
511ffa4698
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 5 次插入1 次删除
  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)

正在加载...
取消
保存