From 265cb22c07f2936d42fa70a28abff34d58de0083 Mon Sep 17 00:00:00 2001 From: Milan Date: Fri, 1 May 2026 01:49:28 +0300 Subject: [PATCH] style(proxy): black-format team member budget refactor files Made-with: Cursor --- litellm/proxy/management_endpoints/common_utils.py | 4 +--- litellm/proxy/management_endpoints/team_endpoints.py | 8 ++------ 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/litellm/proxy/management_endpoints/common_utils.py b/litellm/proxy/management_endpoints/common_utils.py index 2fc1e7cf84d..3d6bebaf909 100644 --- a/litellm/proxy/management_endpoints/common_utils.py +++ b/litellm/proxy/management_endpoints/common_utils.py @@ -363,9 +363,7 @@ def _budget_disconnect_all_unset( ) -async def _disconnect_team_member_budget( - tx, *, team_id: str, user_id: str -) -> None: +async def _disconnect_team_member_budget(tx, *, team_id: str, user_id: str) -> None: await tx.litellm_teammembership.update( where={"user_id_team_id": {"user_id": user_id, "team_id": team_id}}, data={"litellm_budget_table": {"disconnect": True}}, diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py index a53019dddf9..2cf2dc80723 100644 --- a/litellm/proxy/management_endpoints/team_endpoints.py +++ b/litellm/proxy/management_endpoints/team_endpoints.py @@ -2781,9 +2781,7 @@ async def team_member_update( existing_team_row = await _team_member_update_fetch_team_or_raise( prisma_client, data.team_id ) - await _team_member_update_require_authorized( - user_api_key_dict, existing_team_row - ) + await _team_member_update_require_authorized(user_api_key_dict, existing_team_row) returned_team_info: TeamInfoResponseObject = await team_info( http_request=http_request, @@ -4355,9 +4353,7 @@ async def list_team( except Exception as e: team_exception = """Invalid team object for team_id: {}. team_object={}. Error: {} - """.format( - team.team_id, team.model_dump(), str(e) - ) + """.format(team.team_id, team.model_dump(), str(e)) verbose_proxy_logger.exception(team_exception) continue # Sort the responses by team_alias