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

Merge pull request #2049 from BerriAI/litellm_return_all_teams_for_user

Browse source 
feat(proxy_server.py): return all teams, user is a member of in /user/info
This commit is contained in:
Krish Dholakia 2024-02-17 23:36:45 -08:00 • committed by GitHub
commit 9522f6a2c1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 15 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/proxy_server.py
View file

@ -4029,6 +4029,10 @@ async def user_info(
user_info = await prisma_client.get_data(user_id=user_id)
else:
user_info = None
## GET ALL TEAMS ##
teams = await prisma_client.get_data(
user_id=user_id, table_name="team", query_type="find_all"
)
## GET ALL KEYS ##
keys = await prisma_client.get_data(
user_id=user_id,
@ -4052,7 +4056,12 @@ async def user_info(
# if using pydantic v1
key = key.dict()
key.pop("token", None)
return {"user_id": user_id, "user_info": user_info, "keys": keys}
return {
"user_id": user_id,
"user_info": user_info,
"keys": keys,
"teams": teams,
}
except Exception as e:
if isinstance(e, HTTPException):
raise ProxyException(