scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: req changes from greptile

Browse source
This commit is contained in:
Harshit28j 2026-03-22 04:13:50 +05:30
parent 82fecaa9a2
commit 237d8354a0
2 changed files with 21 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -46,6 +46,7 @@ from litellm.proxy.auth.auth_checks import (
can_team_access_model,
get_org_object,
get_project_object,
get_team_membership,
get_team_object,
)
from litellm.proxy.auth.auth_utils import abbreviate_api_key
@ -642,19 +643,17 @@ async def _common_key_generation_helper( # noqa: PLR0915
litellm.team_model_overrides_enabled
or os.getenv("TEAM_MODEL_OVERRIDES", "").lower() == "true"
) and team_table is not None:
# Read member models from LiteLLM_TeamMembership table (authoritative source),
# NOT from members_with_roles JSON blob which can be stale after /team/member_update.
# Note: This is a management endpoint (/key/generate), not the hot auth path.
# The hot path (/chat/completions) uses the SQL view join with zero extra queries.
# Read member models from LiteLLM_TeamMembership via the standard cached helper
# (NOT from members_with_roles JSON blob which can be stale after /team/member_update).
member_models: List[str] = []
if data.user_id and prisma_client is not None:
_membership = await prisma_client.db.litellm_teammembership.find_unique(
where={
"user_id_team_id": {
"user_id": data.user_id,
"team_id": team_table.team_id,
}
}
from litellm.proxy.proxy_server import user_api_key_cache
_membership = await get_team_membership(
user_id=data.user_id,
team_id=team_table.team_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
)
if _membership is not None:
member_models = _membership.models or []

20
litellm/proxy/management_endpoints/team_endpoints.py
View file

@ -2530,17 +2530,19 @@ async def team_member_update( # noqa: PLR0915
### update team member role
# Resolve the effective models for this member (from the authoritative
# LiteLLM_TeamMembership table) so we can: (a) keep the members_with_roles
# JSON in sync, and (b) return the actual stored state in the response.
# LiteLLM_TeamMembership table via cached helper) so we can: (a) keep the
# members_with_roles JSON in sync, and (b) return the actual stored state
# in the response.
stored_models = data.models
if stored_models is None:
_tm_row = await prisma_client.db.litellm_teammembership.find_unique(
where={
"user_id_team_id": {
"user_id": received_user_id,
"team_id": data.team_id,
}
}
from litellm.proxy.auth.auth_checks import get_team_membership
from litellm.proxy.proxy_server import user_api_key_cache
_tm_row = await get_team_membership(
user_id=received_user_id,
team_id=data.team_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
)
stored_models = (_tm_row.models or []) if _tm_row is not None else []