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

fix(team_endpoints.py): always pop team_member_budget from kv

Browse source
This commit is contained in:
Krrish Dholakia 2025-07-20 21:44:34 -07:00
parent 950828e2f5
commit 311c2f47ec
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -209,7 +209,6 @@ async def _upsert_team_member_budget_table(
if updated_kv.get("metadata") is None:
updated_kv["metadata"] = {}
updated_kv["metadata"]["team_member_budget_id"] = budget_row.budget_id
updated_kv.pop("team_member_budget", None)
else: # budget does not exist
updated_kv = await _create_team_member_budget_table(
data=team_table,
@ -217,6 +216,8 @@ async def _upsert_team_member_budget_table(
user_api_key_dict=user_api_key_dict,
team_member_budget=team_member_budget,
)
updated_kv.pop("team_member_budget", None)
return updated_kv
@ -406,7 +407,6 @@ async def new_team( # noqa: PLR0915
_model_id = model_dict.id
## Handle Object Permission - MCP, Vector Stores etc.
object_permission_id = await _set_object_permission(
data=data,