From ed9b03466833d73f03f2daecb973d27447dec01b Mon Sep 17 00:00:00 2001
From: cthomas <caren@letta.com>
Date: Wed, 28 May 2025 23:45:24 -0700
Subject: [PATCH] feat: remove redundant memory rebuild (#2514)

---
 letta/agents/letta_agent.py | 21 +++------------------
 1 file changed, 3 insertions(+), 18 deletions(-)

diff --git a/letta/agents/letta_agent.py b/letta/agents/letta_agent.py
index 5ff43790..09981168 100644
--- a/letta/agents/letta_agent.py
+++ b/letta/agents/letta_agent.py
@@ -128,12 +128,7 @@ class LettaAgent(BaseAgent):
         for _ in range(max_steps):
             step_id = generate_step_id()
 
-            in_context_messages = await self._rebuild_memory_async(
-                current_in_context_messages + new_in_context_messages,
-                agent_state,
-                num_messages=self.num_messages,
-                num_archival_memories=self.num_archival_memories,
-            )
+            in_context_messages = current_in_context_messages + new_in_context_messages
             log_event("agent.stream_no_tokens.messages.refreshed")  # [1^]
 
             request_data = await self._create_llm_request_data_async(
@@ -236,12 +231,7 @@ class LettaAgent(BaseAgent):
         for _ in range(max_steps):
             step_id = generate_step_id()
 
-            in_context_messages = await self._rebuild_memory_async(
-                current_in_context_messages + new_in_context_messages,
-                agent_state,
-                num_messages=self.num_messages,
-                num_archival_memories=self.num_archival_memories,
-            )
+            in_context_messages = current_in_context_messages + new_in_context_messages
             log_event("agent.step.messages.refreshed")  # [1^]
 
             request_data = await self._create_llm_request_data_async(
@@ -347,12 +337,7 @@ class LettaAgent(BaseAgent):
         provider_request_start_timestamp_ns = None
         for _ in range(max_steps):
             step_id = generate_step_id()
-            in_context_messages = await self._rebuild_memory_async(
-                current_in_context_messages + new_in_context_messages,
-                agent_state,
-                num_messages=self.num_messages,
-                num_archival_memories=self.num_archival_memories,
-            )
+            in_context_messages = current_in_context_messages + new_in_context_messages
             log_event("agent.step.messages.refreshed")  # [1^]
 
             request_data = await self._create_llm_request_data_async(