From 22b640a5dd14cffac6d7f6901916a3e9897264f6 Mon Sep 17 00:00:00 2001 From: Shangyin Tan Date: Thu, 12 Jun 2025 17:11:36 -0700 Subject: [PATCH] fix: adjust memory summary's recall message count (#2778) --- letta/agents/base_agent.py | 2 +- letta/services/agent_manager.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/letta/agents/base_agent.py b/letta/agents/base_agent.py index d4e4e38f..9624fa49 100644 --- a/letta/agents/base_agent.py +++ b/letta/agents/base_agent.py @@ -117,7 +117,7 @@ class BaseAgent(ABC): system_prompt=agent_state.system, in_context_memory=agent_state.memory, in_context_memory_last_edit=memory_edit_timestamp, - previous_message_count=num_messages, + previous_message_count=num_messages - len(in_context_messages), archival_memory_size=num_archival_memories, tool_rules_solver=tool_rules_solver, ) diff --git a/letta/services/agent_manager.py b/letta/services/agent_manager.py index a100c5cd..dc292d1f 100644 --- a/letta/services/agent_manager.py +++ b/letta/services/agent_manager.py @@ -1419,7 +1419,7 @@ class AgentManager: system_prompt=agent_state.system, in_context_memory=agent_state.memory, in_context_memory_last_edit=memory_edit_timestamp, - previous_message_count=num_messages, + previous_message_count=num_messages - len(agent_state.message_ids), archival_memory_size=num_archival_memories, ) @@ -1493,7 +1493,7 @@ class AgentManager: system_prompt=agent_state.system, in_context_memory=agent_state.memory, in_context_memory_last_edit=memory_edit_timestamp, - previous_message_count=num_messages, + previous_message_count=num_messages - len(agent_state.message_ids), archival_memory_size=num_archival_memories, tool_rules_solver=tool_rules_solver, )