From 311c2f47ecfdb64d4ec336a041015a2b4d7a5430 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sun, 20 Jul 2025 21:44:34 -0700 Subject: [PATCH] fix(team_endpoints.py): always pop team_member_budget from kv --- litellm/proxy/management_endpoints/team_endpoints.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py index b48d063b708..1ba363d1fdb 100644 --- a/litellm/proxy/management_endpoints/team_endpoints.py +++ b/litellm/proxy/management_endpoints/team_endpoints.py @@ -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,