phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix linting errors when adding a new team member

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-24 16:38:43 -07:00
parent 0ad5f58930
commit 20840eaad3
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -571,6 +571,11 @@ async def team_member_add(
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
# Check if updated_team is None
if updated_team is None:
raise HTTPException(
status_code=404, detail={"error": f"Team with id {data.team_id} not found"}
)
return TeamAddMemberResponse( return TeamAddMemberResponse(
**updated_team.model_dump(), **updated_team.model_dump(),
updated_users=updated_users, updated_users=updated_users,

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

@ -103,7 +103,7 @@ async def add_new_member(
elif len(existing_user_row) == 1: elif len(existing_user_row) == 1:
user_info = existing_user_row[0] user_info = existing_user_row[0]
_returned_user = await prisma_client.db.litellm_usertable.update( _returned_user = await prisma_client.db.litellm_usertable.update(
where={"user_id": user_info.user_id}, where={"user_id": user_info.user_id}, # type: ignore
data={"teams": {"push": [team_id]}}, data={"teams": {"push": [team_id]}},
) )
returned_user = LiteLLM_UserTable(**_returned_user.model_dump()) returned_user = LiteLLM_UserTable(**_returned_user.model_dump())