scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

refactor(proxy): resolve MCP display names for 403 detail in a single DB lookup

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-15 13:09:16 +00:00
parent 8992229697
commit 79c534ecec
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/management_helpers/object_permission_utils.py
View file

@ -300,10 +300,10 @@ async def _resolve_mcp_server_identifiers_to_ids(
return resolved
async def _mcp_server_display_names(
async def _mcp_server_display_name_map(
server_ids: AbstractSet[str],
prisma_client: PrismaClient | None,
) -> list[str]:
) -> Mapping[str, str]:
"""
Map MCP server IDs to human-readable names for error messages.
@ -331,8 +331,7 @@ async def _mcp_server_display_names(
)
}
)
id_to_name: Final = MappingProxyType({**registry_names, **db_names})
return sorted(id_to_name.get(server_id, server_id) for server_id in server_ids)
return MappingProxyType({**registry_names, **db_names})
_MCP_TOOL_PERMISSIONS_ADAPTER: Final = TypeAdapter(dict[str, list[str] | None])
@ -729,18 +728,19 @@ async def validate_key_mcp_servers_against_team(
)
disallowed_servers: Final = active_requested_servers - allowed_servers - grandfathered_servers
if disallowed_servers:
disallowed_names: Final = await _mcp_server_display_names(
server_ids=disallowed_servers,
display_names: Final = await _mcp_server_display_name_map(
server_ids=disallowed_servers | team_allowed_servers | allow_all_keys_servers,
prisma_client=prisma_client,
)
allow_all_names: Final = await _mcp_server_display_names(
server_ids=allow_all_keys_servers,
prisma_client=prisma_client,
disallowed_names: Final = sorted(
display_names.get(server_id, server_id) for server_id in disallowed_servers
)
allow_all_names: Final = sorted(
display_names.get(server_id, server_id) for server_id in allow_all_keys_servers
)
if team_obj is not None:
team_allowed_names: Final = await _mcp_server_display_names(
server_ids=team_allowed_servers,
prisma_client=prisma_client,
team_allowed_names: Final = sorted(
display_names.get(server_id, server_id) for server_id in team_allowed_servers
)
team_display: Final = team_obj.team_alias or team_obj.team_id
detail = (