From 7f793246c21583240d65bb89bb030ec08aa7f2a9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 28 Mar 2024 09:41:15 -0700 Subject: [PATCH] (feat) return team alias when hitting user/info --- litellm/proxy/proxy_server.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 73cbd55e43..be6c12db1e 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -5290,6 +5290,18 @@ async def user_info( key = key.dict() key.pop("token", None) + if ( + "team_id" in key + and key["team_id"] is not None + and key["team_id"] != "litellm-dashboard" + ): + team_info = await prisma_client.get_data( + team_id=key["team_id"], table_name="team" + ) + key["team_alias"] = team_info["team_alias"] + else: + key["team_alias"] = "None" + response_data = { "user_id": user_id, "user_info": user_info,