diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py index a709af811b2..ab4ab48de51 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py @@ -627,6 +627,7 @@ async def test_get_tools_from_mcp_servers_continues_when_one_server_fails(): extra_headers=None, add_prefix=True, raw_headers=None, + **kwargs, ): if server.name == "working_server": # Working server returns tools @@ -732,6 +733,7 @@ async def test_get_tools_from_mcp_servers_handles_all_servers_failing(): extra_headers=None, add_prefix=True, raw_headers=None, + **kwargs, ): # All servers fail raise Exception(f"Server {server.name} connection failed") @@ -1173,6 +1175,7 @@ async def test_list_tools_single_server_unprefixed_names(): extra_headers=None, add_prefix=False, raw_headers=None, + **kwargs, ): tool = MagicMock() tool.name = f"{server.alias}-toolA" if add_prefix else "toolA" @@ -1254,6 +1257,7 @@ async def test_list_tools_multiple_servers_prefixed_names(): extra_headers=None, add_prefix=True, raw_headers=None, + **kwargs, ): tool = MagicMock() # When multiple servers, add_prefix should be True -> prefixed names @@ -1520,6 +1524,7 @@ async def test_list_tools_filters_by_key_team_permissions(): extra_headers=None, add_prefix=False, raw_headers=None, + **kwargs, ): # Return 4 tools, but only 2 should be allowed tool1 = MagicMock() @@ -1629,6 +1634,7 @@ async def test_list_tools_with_team_tool_permissions_inheritance(): extra_headers=None, add_prefix=False, raw_headers=None, + **kwargs, ): # Return 4 tools tool1 = MagicMock() @@ -1724,6 +1730,7 @@ async def test_list_tools_with_no_tool_permissions_shows_all(): extra_headers=None, add_prefix=False, raw_headers=None, + **kwargs, ): # Return 3 tools tool1 = MagicMock() @@ -1822,6 +1829,7 @@ async def test_list_tools_strips_prefix_when_matching_permissions(): extra_headers=None, add_prefix=True, raw_headers=None, + **kwargs, ): # Return tools WITH prefix (as they come from MCP server) tool1 = MagicMock()