Skip to content

Commit

Permalink
fix: follow-up (suggested questions) does not refer to the most recen… (
Browse files Browse the repository at this point in the history
  • Loading branch information
xuzuodong authored Oct 9, 2024
1 parent 7a1da24 commit 511ffa4
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion api/core/memory/token_buffer_memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,11 @@ def get_history_prompt_messages(
# 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)
Expand Down

0 comments on commit 511ffa4

Please sign in to comment.