scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

refactor(team): derive bulk update response fields from budget patch map

Browse source 
Drop the duplicate frozenset allowlist and reuse _MEMBER_BUDGET_PATCH_FIELDS
so successful_updates stays in sync with the budget fields the write path applies
This commit is contained in:
mubashir1osmani 2026-07-16 17:11:57 -07:00
parent a3768b489f
commit 1788560aef
1 changed files with 1 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -3074,17 +3074,6 @@ async def _apply_team_member_update(
)
_BULK_MEMBER_RESPONSE_FIELDS = frozenset(
{
"max_budget_in_team",
"tpm_limit",
"rpm_limit",
"budget_duration",
"allowed_models",
}
)
def _successful_member_update_response(
*,
team_id: str,
@ -3095,7 +3084,7 @@ def _successful_member_update_response(
return TeamMemberUpdateResponse(
team_id=team_id,
user_id=user_id,
**{field: getattr(update_fields, field) for field in _BULK_MEMBER_RESPONSE_FIELDS if field in fields_set},
**{field: getattr(update_fields, field) for field in _MEMBER_BUDGET_PATCH_FIELDS if field in fields_set},
)