scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

resolved greptile issue

Browse source
This commit is contained in:
shivam 2026-02-27 03:36:43 -08:00
parent c090941e4a
commit 268a744810
1 changed files with 16 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -4,9 +4,11 @@ organizations, teams, and keys.
"""
import json
from litellm._uuid import uuid
from typing import TYPE_CHECKING, Any, Dict, Optional, Union
from fastapi import HTTPException, status
from litellm._uuid import uuid
if TYPE_CHECKING:
from litellm.proxy._types import UserAPIKeyAuth
@ -220,15 +222,19 @@ async def get_allowed_mcp_access_groups_for_user(
for auth_context in auth_contexts:
if auth_context.team_id:
team_obj = await get_team_object(
team_id=auth_context.team_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
parent_otel_span=getattr(
user_api_key_dict, "parent_otel_span", None
),
proxy_logging_obj=proxy_logging_obj,
)
try:
team_obj = await get_team_object(
team_id=auth_context.team_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
parent_otel_span=getattr(
user_api_key_dict, "parent_otel_span", None
),
proxy_logging_obj=proxy_logging_obj,
)
except HTTPException:
# Team not found (e.g. deleted) - skip gracefully
continue
if team_obj and team_obj.object_permission:
groups = team_obj.object_permission.mcp_access_groups or []
allowed_access_group_ids.update(groups)
@ -254,8 +260,6 @@ async def validate_mcp_object_permission_for_key(
HTTPException: 403 if user tries to assign MCP servers or access groups
they do not have access to.
"""
from fastapi import HTTPException, status
from litellm.proxy.management_endpoints.common_utils import _user_has_admin_view
if object_permission is None: