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

refactor(proxy): extract soft team resolve to restore C901 budget

Browse source 
Move existing-key team soft-resolve (404-tolerant) out of
_process_single_key_update so McCabe stays within the strict C901 ceiling.
This commit is contained in:
leilei3167 2026-09-16 16:15:00 +00:00
parent 9771996d04
commit a25d2926b0
1 changed files with 30 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1574,6 +1574,28 @@ def _maybe_add_key_team_limit_warnings(
) # mutable-ok: GenerateKeyResponse/tests expect list warnings
async def _soft_resolve_existing_key_team_for_warnings(
existing_key_row: LiteLLM_VerificationToken,
prisma_client: PrismaClient | None,
user_api_key_cache: UserApiKeyCache,
) -> LiteLLM_TeamTableCachedObj | None:
"""Resolve the existing key's team for warning payloads only; ignore missing teams."""
team_id = getattr(existing_key_row, "team_id", None)
if team_id is None or prisma_client is None:
return None
try:
return await get_team_object(
team_id=team_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
check_db_only=True,
)
except HTTPException as e:
if e.status_code != status.HTTP_404_NOT_FOUND:
raise
return None
async def _check_team_key_limits(
team_table: LiteLLM_TeamTableCachedObj,
data: GenerateKeyRequest | UpdateKeyRequest,
@ -2622,8 +2644,8 @@ async def _process_single_key_update(
_enforce_upperbound_key_params(update_key_request, fill_defaults=False)
# Get team object and check team limits if team_id is provided on the request.
# Existing-key team is soft-resolved below for warnings only — a missing team
# must not block an otherwise valid update (custom key policy runs later).
# Existing-key team is soft-resolved for warnings only — a missing team must not
# block an otherwise valid update (custom key policy runs later).
team_obj: LiteLLM_TeamTableCachedObj | None = None
if update_key_request.team_id is not None:
team_obj = await get_team_object(
@ -2639,18 +2661,12 @@ async def _process_single_key_update(
data=update_key_request,
prisma_client=prisma_client,
)
elif getattr(existing_key_row, "team_id", None) is not None and prisma_client is not None:
try:
team_obj = await get_team_object(
team_id=existing_key_row.team_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
check_db_only=True,
)
except HTTPException as e:
if e.status_code != status.HTTP_404_NOT_FOUND:
raise
team_obj = None
else:
team_obj = await _soft_resolve_existing_key_team_for_warnings(
existing_key_row=existing_key_row,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
)
# Validate team change if team is being changed
if is_different_team(data=update_key_request, existing_key_row=existing_key_row):