letta-nightly 0.8.15.dev20250720104313__py3-none-any.whl → 0.8.16.dev20250721070720__py3-none-any.whl

letta/__init__.py

@@ -5,7 +5,7 @@ try:
     __version__ = version("letta")
 except PackageNotFoundError:
     # Fallback for development installations
-    __version__ = "0.8.15"
+    __version__ = "0.8.16"
 
 if os.environ.get("LETTA_VERSION"):
     __version__ = os.environ["LETTA_VERSION"]

letta/agent.py

@@ -1304,12 +1304,11 @@ class Agent(BaseAgent):
     async def get_context_window_from_tiktoken_async(self) -> ContextWindowOverview:
         """Get the context window of the agent"""
         # Grab the in-context messages
-        # conversion of messages to OpenAI dict format, which is passed to the token counter
-        (in_context_messages, passage_manager_size, message_manager_size) = await asyncio.gather(
-            self.message_manager.get_messages_by_ids_async(message_ids=self.agent_state.message_ids, actor=self.user),
-            self.passage_manager.agent_passage_size_async(actor=self.user, agent_id=self.agent_state.id),
-            self.message_manager.size_async(actor=self.user, agent_id=self.agent_state.id),
+        in_context_messages = await self.message_manager.get_messages_by_ids_async(
+            message_ids=self.agent_state.message_ids, actor=self.user
         )
+
+        # conversion of messages to OpenAI dict format, which is passed to the token counter
         in_context_messages_openai = [m.to_openai_dict() for m in in_context_messages]
 
         # Extract system, memory and external summary
@@ -1396,6 +1395,15 @@ class Agent(BaseAgent):
         )
         assert isinstance(num_tokens_used_total, int)
 
+        passage_manager_size = await self.passage_manager.agent_passage_size_async(
+            agent_id=self.agent_state.id,
+            actor=self.user,
+        )
+        message_manager_size = await self.message_manager.size_async(
+            agent_id=self.agent_state.id,
+            actor=self.user,
+        )
+
         return ContextWindowOverview(
             # context window breakdown (in messages)
             num_messages=len(in_context_messages),
@@ -1426,12 +1434,11 @@ class Agent(BaseAgent):
         model = self.agent_state.llm_config.model if self.agent_state.llm_config.model_endpoint_type == "anthropic" else None
 
         # Grab the in-context messages
-        # conversion of messages to anthropic dict format, which is passed to the token counter
-        (in_context_messages, passage_manager_size, message_manager_size) = await asyncio.gather(
-            self.message_manager.get_messages_by_ids_async(message_ids=self.agent_state.message_ids, actor=self.user),
-            self.passage_manager.agent_passage_size_async(actor=self.user, agent_id=self.agent_state.id),
-            self.message_manager.size_async(actor=self.user, agent_id=self.agent_state.id),
+        in_context_messages = await self.message_manager.get_messages_by_ids_async(
+            message_ids=self.agent_state.message_ids, actor=self.user
         )
+
+        # conversion of messages to anthropic dict format, which is passed to the token counter
         in_context_messages_anthropic = [m.to_anthropic_dict() for m in in_context_messages]
 
         # Extract system, memory and external summary
@@ -1546,6 +1553,15 @@ class Agent(BaseAgent):
         )
         assert isinstance(num_tokens_used_total, int)
 
+        passage_manager_size = await self.passage_manager.agent_passage_size_async(
+            agent_id=self.agent_state.id,
+            actor=self.user,
+        )
+        message_manager_size = await self.message_manager.size_async(
+            agent_id=self.agent_state.id,
+            actor=self.user,
+        )
+
         return ContextWindowOverview(
             # context window breakdown (in messages)
             num_messages=len(in_context_messages),
@@ -1625,7 +1641,7 @@ class Agent(BaseAgent):
                 # Get the MCPClient from the server's handle
                 # TODO these don't get raised properly
                 if not self.mcp_clients:
-                    raise ValueError(f"No MCP client available to use")
+                    raise ValueError("No MCP client available to use")
                 if server_name not in self.mcp_clients:
                     raise ValueError(f"Unknown MCP server name: {server_name}")
                 mcp_client = self.mcp_clients[server_name]

letta/agents/helpers.py

@@ -235,5 +235,5 @@ def _pop_heartbeat(tool_args: dict) -> bool:
 def _build_rule_violation_result(tool_name: str, valid: list[str], solver: ToolRulesSolver) -> ToolExecutionResult:
     hint_lines = solver.guess_rule_violation(tool_name)
     hint_txt = ("\n** Hint: Possible rules that were violated:\n" + "\n".join(f"\t- {h}" for h in hint_lines)) if hint_lines else ""
-    msg = f"[ToolConstraintError] Cannot call {tool_name}, " f"valid tools include: {valid}.{hint_txt}"
+    msg = f"[ToolConstraintError] Cannot call {tool_name}, valid tools include: {valid}.{hint_txt}"
     return ToolExecutionResult(status="error", func_return=msg)