scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

fix(mcp): update lazy openapi snapshot and reduce get_mcp_access_groups complexity

Browse source
This commit is contained in:
Pyhroff 2026-10-28 20:42:55 +05:30
parent 91ac9a5b7d
commit 86beb0403a
2 changed files with 15 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/_lazy_openapi_snapshot.json
View file

@ -42714,7 +42714,7 @@
"paths": {
"/v1/mcp/access_groups": {
"get": {
"description": "Get all available MCP access groups from the database AND config",
"description": "Get available MCP access groups from the database AND config.\nAdmins receive the full catalog; non-admin callers are scoped to servers they have access to.",
"operationId": "get_mcp_access_groups_v1_mcp_access_groups_get",
"responses": {
"200": {

39
litellm/proxy/management_endpoints/mcp_management_endpoints.py
View file

@ -1122,42 +1122,31 @@ if MCP_AVAILABLE:
access_groups: Final[set[str]] = set()
if user_api_key_has_admin_view(user_api_key_dict):
# Get from config-loaded servers
for server in global_mcp_server_manager.config_mcp_servers.values():
cfg_groups = getattr(server, "access_groups", None)
if cfg_groups and isinstance(cfg_groups, (list, set, tuple)):
for g in cfg_groups:
def _collect_groups(server_obj: Any) -> None:
for attr in ("mcp_access_groups", "access_groups"):
raw_groups = getattr(server_obj, attr, None)
if raw_groups and isinstance(raw_groups, (list, set, tuple)):
for g in raw_groups:
if g:
access_groups.add(str(g))
# Get from DB
if user_api_key_has_admin_view(user_api_key_dict):
for cfg_server in global_mcp_server_manager.config_mcp_servers.values():
_collect_groups(cfg_server)
if prisma_client is not None:
try:
mcp_servers: Final[Sequence[prisma_models.LiteLLM_MCPServerTable]] = await MCPServerRepository(
db_servers: Final[Sequence[prisma_models.LiteLLM_MCPServerTable]] = await MCPServerRepository(
prisma_client
).table.find_many()
for server in mcp_servers:
mcp_groups = getattr(server, "mcp_access_groups", None)
if mcp_groups and isinstance(mcp_groups, (list, set, tuple)):
for g in mcp_groups:
if g:
access_groups.add(str(g))
for db_server in db_servers:
_collect_groups(db_server)
except Exception as e:
verbose_proxy_logger.debug("Error getting MCP access groups: %s", e)
else:
accessible_servers: Final = await _resolve_accessible_mcp_servers(user_api_key_dict)
for server in accessible_servers:
mcp_groups = getattr(server, "mcp_access_groups", None)
if mcp_groups and isinstance(mcp_groups, (list, set, tuple)):
for g in mcp_groups:
if g:
access_groups.add(str(g))
cfg_groups = getattr(server, "access_groups", None)
if cfg_groups and isinstance(cfg_groups, (list, set, tuple)):
for g in cfg_groups:
if g:
access_groups.add(str(g))
for acc_server in accessible_servers:
_collect_groups(acc_server)
# Convert to sorted list
access_groups_list: Final[list[str]] = sorted(list(access_groups))