From 09113078c02c111e61e91efe9ae006ceab70f09a Mon Sep 17 00:00:00 2001 From: Ryan Crabbe Date: Thu, 23 Apr 2026 16:00:28 -0700 Subject: [PATCH] style: apply black formatting --- litellm/proxy/_experimental/mcp_server/rest_endpoints.py | 8 +++++--- .../_experimental/mcp_server/test_mcp_server_manager.py | 4 +++- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/_experimental/mcp_server/rest_endpoints.py b/litellm/proxy/_experimental/mcp_server/rest_endpoints.py index f1030bf0a8..3047fb7332 100644 --- a/litellm/proxy/_experimental/mcp_server/rest_endpoints.py +++ b/litellm/proxy/_experimental/mcp_server/rest_endpoints.py @@ -317,9 +317,11 @@ if MCP_AVAILABLE: ): # Dict keys may be server_ids OR names/aliases; normalize so lookup # by concrete server_id resolves name-keyed restrictions too. - allowed_tools_for_server = global_mcp_server_manager.expand_tool_permissions( - user_api_key_auth.object_permission.mcp_tool_permissions - ).get(server.server_id) + allowed_tools_for_server = ( + global_mcp_server_manager.expand_tool_permissions( + user_api_key_auth.object_permission.mcp_tool_permissions + ).get(server.server_id) + ) if ( allowed_tools_for_server is not None and len(allowed_tools_for_server) > 0 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 6ac2b0ac4f..447c28078e 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 @@ -2673,7 +2673,9 @@ class TestMCPServerManagerExpandPermissionList: def test_searches_config_and_registry_union(self): manager = MCPServerManager() - manager.config_mcp_servers["cfg-id"] = self._make_server("cfg-id", server_name="a") + manager.config_mcp_servers["cfg-id"] = self._make_server( + "cfg-id", server_name="a" + ) manager.registry["reg-id"] = self._make_server("reg-id", server_name="b") assert manager.expand_permission_list(["a"]) == ["cfg-id"]