ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: apply black formatting

Browse Source
This commit is contained in:
Ryan Crabbe 2026-04-23 16:00:28 -07:00
parent 57b0d7f45f
commit 09113078c0
No known key found for this signature in database
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
litellm/proxy/_experimental/mcp_server/rest_endpoints.py
View File

@ -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

4
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
View File

@ -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"]