scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

style(proxy): black-format team member budget refactor files

Browse source 
Made-with: Cursor
This commit is contained in:
Milan 2026-05-01 01:49:28 +03:00
parent e726e910df
commit 265cb22c07
No known key found for this signature in database
2 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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}},

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

@ -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