phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(proxy admin ui) - show Teams sorted by Team Alias (#7296)

Browse source 
* ui - sort teams by team alias

* test assert /user/info returns teams in a sorted order

* fix team_alias check on team
This commit is contained in:
Ishaan Jaff 2024-12-18 19:43:19 -08:00 • committed by GitHub
parent 6220e17ebf
commit 087ab41d66
3 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/management_endpoints/internal_user_endpoints.py
View file

@ -382,7 +382,7 @@ async def user_info(
## REMOVE HASHED TOKEN INFO before returning ##
returned_keys = _process_keys_for_user_info(keys=keys, all_teams=teams_1)
team_list.sort(key=lambda x: (getattr(x, "team_alias", "")))
_user_info = (
user_info.model_dump() if isinstance(user_info, BaseModel) else user_info
)
@ -436,6 +436,7 @@ async def _get_user_info_for_proxy_admin():
# cast all teams to LiteLLM_TeamTable
_teams_in_db: List = results[0]["teams"] or []
_teams_in_db = [LiteLLM_TeamTable(**team) for team in _teams_in_db]
_teams_in_db.sort(key=lambda x: (getattr(x, "team_alias", "") or ""))
returned_keys = _process_keys_for_user_info(keys=keys_in_db, all_teams=_teams_in_db)
return UserInfoResponse(
user_id=None,