From 18c9e8910685fefee2fb5f67e7fdd1cb37b67750 Mon Sep 17 00:00:00 2001
From: teknium1 <127238744+teknium1@users.noreply.github.com>
Date: Fri, 29 May 2026 01:32:18 -0700
Subject: [PATCH] test: update _invoke_tool dispatch assertion for new
 toolset-scope kwargs

The scoping fix added enabled_toolsets/disabled_toolsets to the
agent_runtime_helpers sequential dispatch into handle_function_call, so
test_invoke_tool_dispatches_to_handle_function_call's assert_called_once_with
(exact match) needs the two new kwargs. Both are None for the default agent
fixture.
---
 tests/run_agent/test_run_agent.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/run_agent/test_run_agent.py b/tests/run_agent/test_run_agent.py
index b963965c4a5..20aec422cf1 100644
--- a/tests/run_agent/test_run_agent.py
+++ b/tests/run_agent/test_run_agent.py
@@ -2428,6 +2428,8 @@ class TestConcurrentToolExecution:
                 session_id=agent.session_id,
                 enabled_tools=list(agent.valid_tool_names),
                 skip_pre_tool_call_hook=True,
+                enabled_toolsets=agent.enabled_toolsets,
+                disabled_toolsets=agent.disabled_toolsets,
             )
             assert result == "result"