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

feat(proxy_server.py): support retrieving models for a team, if user is a member - via /models?team_id

Browse source 
Allows user to see team models on UI when creating a key
This commit is contained in:
Krrish Dholakia 2025-03-14 19:34:06 -07:00
parent 1abce5c47b
commit fe148e3dec
4 changed files with 32 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/management_endpoints/team_endpoints.py
View file

@ -1352,9 +1352,9 @@ async def team_info(
else:
_team_info = LiteLLM_TeamTable()
## UNFURL 'all-proxy-models' into the team_info.models list ##
if llm_router is not None:
_team_info = _unfurl_all_proxy_models(_team_info, llm_router)
# ## UNFURL 'all-proxy-models' into the team_info.models list ##
# if llm_router is not None:
# _team_info = _unfurl_all_proxy_models(_team_info, llm_router)
response_object = TeamInfoResponseObject(
team_id=team_id,
team_info=_team_info,
@ -1615,11 +1615,6 @@ async def list_team(
)
try:
# unfurl all-proxy-models
if llm_router is not None:
team = _unfurl_all_proxy_models(
LiteLLM_TeamTable(**team.model_dump()), llm_router
)
returned_responses.append(
TeamListResponseObject(
**team.model_dump(),