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

feat: parallel tool calling in model settings [LET-6239] (#6262)

Browse Source 
* parallel tool calling in model settings

* configs for send message sdk v1

* change models for all tests

---------

Co-authored-by: Ari Webb <ari@letta.com>
This commit is contained in:
Ari Webb
2025-11-20 13:00:31 -08:00
committed by Caren Thomas
parent 2d1667a97c
commit d417870537
7 changed files with 28 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/sdk_v1/integration/integration_test_send_message_v2.py
View File

@@ -644,17 +644,14 @@ async def test_parallel_tool_calls(
if provider_type in ["google_ai", "google_vertex"]:
pytest.skip("Gemini models are flaky for this test so we disable them for now")
# # Update model_settings to enable parallel tool calling
# modified_model_settings = model_settings.copy()
# modified_model_settings["parallel_tool_calls"] = True
# Update model_settings to enable parallel tool calling
modified_model_settings = model_settings.copy()
modified_model_settings["parallel_tool_calls"] = True
# IMPORTANT: Set parallel_tool_calls at BOTH the agent level and in model_settings
# Even though the agent-level parameter is deprecated, it may still be needed
agent_state = await client.agents.update(
agent_id=agent_state.id,
model=model_handle,
model_settings=model_settings,
parallel_tool_calls=True, # Set at agent level as well
model_settings=modified_model_settings,
)
if send_type == "step":