From 46cd428b65f92a445ae18b9ce51996219af8e9d0 Mon Sep 17 00:00:00 2001 From: yucheng Date: Sat, 26 Sep 2026 23:43:36 +0000 Subject: [PATCH] test(mcp): assert the BYOK listing lands in the caller's listed-tool slot Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../proxy/_experimental/mcp_server/test_mcp_server_manager.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py index 3683d00d2a3..e44c66f6e35 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py @@ -7088,6 +7088,9 @@ class TestMCPServerManager: user_api_key_auth=UserAPIKeyAuth(api_key="sk-litellm"), ) + caller: Final = ListedToolsCaller(mcp_auth_header="Bearer hdr") + listed = manager.get_listed_tool(server, "turn", caller) + assert listed is not None and listed.description == "t" assert manager._create_mcp_client.await_args.kwargs["mcp_auth_header"] == "Bearer hdr" @pytest.mark.parametrize(