Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: get rid of client injection in args and just make it always available (#6459)

Browse Source
This commit is contained in:
Sarah Wooders
2025-11-30 00:32:43 -08:00
committed by Caren Thomas
parent f417e53638
commit 86023db9b1
5 changed files with 42 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/integration_test_tool_execution_sandbox.py
View File

@@ -406,22 +406,22 @@ async def test_tool_with_client_injection(disable_e2b_api_key, server: SyncServe
"""Test that tools can access injected letta_client and agent_id to modify agent blocks."""
# Create a tool that uses the injected client and agent_id to actually clear a memory block
# Note: `client` is always available as a variable in the sandbox scope
memory_clear_source = '''
def memory_clear(label: str, agent_id: str, client: "Letta"):
def memory_clear(label: str, agent_id: str):
"""Test tool that clears a memory block using the injected client.
Args:
label: The label of the memory block to clear
agent_id: The agent's ID (injected by Letta system)
client: The Letta client instance (injected by Letta system)
"""
# Verify that agent_id was injected
if not agent_id or not isinstance(agent_id, str):
return f"ERROR: agent_id not properly injected: {agent_id}"
# Verify that client was injected
# Verify that client is available in scope (always injected)
if not client or not hasattr(client, 'agents'):
return f"ERROR: client not properly injected: {client}"
return f"ERROR: client not available in scope: {client}"
# Use the injected client to actually clear the memory block
try:
@@ -504,8 +504,8 @@ def memory_clear(label: str, agent_id: str, client: "Letta"):
# Initialize the sandbox to detect reserved keywords
await sandbox._init_async()
# Verify that the tool correctly detects the need for injection
assert sandbox.inject_letta_client == True # Should detect 'client' parameter
# Verify that injection is configured correctly
assert sandbox.inject_letta_client == True # Client is always injected
assert sandbox.inject_agent_id == True # Should detect 'agent_id' parameter
# Generate the execution script to verify injection code is present