From 0326f959bcb24e28e241ffdc95b4422e04be005a Mon Sep 17 00:00:00 2001 From: cthomas Date: Wed, 28 May 2025 23:29:18 -0700 Subject: [PATCH] feat: remove extra size load (#2513) --- letta/agents/letta_agent.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/letta/agents/letta_agent.py b/letta/agents/letta_agent.py index 822b52e8..1120ae04 100644 --- a/letta/agents/letta_agent.py +++ b/letta/agents/letta_agent.py @@ -75,8 +75,8 @@ class LettaAgent(BaseAgent): self.last_function_response = None # Cached archival memory/message size - self.num_messages = 0 - self.num_archival_memories = 0 + self.num_messages = None + self.num_archival_memories = None self.summarization_agent = None self.summary_block_label = summary_block_label @@ -467,11 +467,6 @@ class LettaAgent(BaseAgent): if not agent_state.message_buffer_autoclear: await self._rebuild_context_window(in_context_messages=current_in_context_messages, new_letta_messages=new_in_context_messages) - # TODO: This may be out of sync, if in between steps users add files - # NOTE (cliandy): temporary for now for particlar use cases. - self.num_messages = await self.message_manager.size_async(actor=self.actor, agent_id=agent_state.id) - self.num_archival_memories = await self.passage_manager.size_async(actor=self.actor, agent_id=agent_state.id) - # TODO: Also yield out a letta usage stats SSE yield f"data: {usage.model_dump_json()}\n\n" yield f"data: {MessageStreamStatus.done.model_dump_json()}\n\n"