phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(team_endpoints.py): return team member budgets in /team/info call

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/5390
This commit is contained in:
Krrish Dholakia 2024-08-28 19:14:01 -07:00
parent f0fb8bdf45
commit 18689c25e9
5 changed files with 53 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/management_helpers/utils.py
View file

@ -120,7 +120,11 @@ async def add_new_member(
)
# Check if trying to set a budget for team member
if max_budget_in_team is not None and new_member.user_id is not None:
if (
max_budget_in_team is not None
and returned_user is not None
and returned_user.user_id is not None
):
# create a new budget item for this member
response = await prisma_client.db.litellm_budgettable.create(
data={
@ -135,7 +139,7 @@ async def add_new_member(
await prisma_client.db.litellm_teammembership.create(
data={
"team_id": team_id,
"user_id": new_member.user_id,
"user_id": returned_user.user_id,
"budget_id": _budget_id,
},
include={"litellm_budget_table": True},