From beb72f94e073a1a19f955d823117cb5c66098e17 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 28 Mar 2024 10:16:07 -0700 Subject: [PATCH] (feat) return team alias in keys --- litellm/proxy/proxy_server.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index be6c12db1e..2fe4a3523d 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -5282,6 +5282,7 @@ async def user_info( user_info = {"spend": spend} ## REMOVE HASHED TOKEN INFO before returning ## + returned_keys = [] for key in keys: try: key = key.model_dump() # noqa @@ -5298,14 +5299,16 @@ async def user_info( team_info = await prisma_client.get_data( team_id=key["team_id"], table_name="team" ) - key["team_alias"] = team_info["team_alias"] + team_alias = getattr(team_info, "team_alias", None) + key["team_alias"] = team_alias else: key["team_alias"] = "None" + returned_keys.append(key) response_data = { "user_id": user_id, "user_info": user_info, - "keys": keys, + "keys": returned_keys, "teams": team_list, } return response_data