diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index be6c12db1..2fe4a3523 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