diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 6642fe80163..c6a67774d14 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -12641,15 +12641,6 @@ async def model_info_v1( # noqa: PLR0915 for model in all_models ] - from litellm.proxy.agent_endpoints.model_list_helpers import ( - append_agents_to_model_info, - ) - - all_models = await append_agents_to_model_info( - models=all_models, - user_api_key_dict=user_api_key_dict, - ) - verbose_proxy_logger.debug("all_models: %s", all_models) return {"data": all_models} diff --git a/tests/test_litellm/proxy/proxy_server/test_team_model_name_translation.py b/tests/test_litellm/proxy/proxy_server/test_team_model_name_translation.py index c6a15eb6d46..570e5b682cf 100644 --- a/tests/test_litellm/proxy/proxy_server/test_team_model_name_translation.py +++ b/tests/test_litellm/proxy/proxy_server/test_team_model_name_translation.py @@ -170,13 +170,6 @@ async def test_model_info_v1_list_path_translates_team_model_name(monkeypatch): monkeypatch.setattr( ps, "_enrich_model_info_with_litellm_data", lambda model, **kw: model ) - import litellm.proxy.agent_endpoints.model_list_helpers as mlh - - monkeypatch.setattr( - mlh, - "append_agents_to_model_info", - AsyncMock(side_effect=lambda models, **kw: models), - ) admin = UserAPIKeyAuth( user_id="u", user_role=LitellmUserRoles.PROXY_ADMIN, team_models=[] @@ -207,13 +200,6 @@ async def test_model_info_v1_unrestricted_key_returns_all_deployments(monkeypatc monkeypatch.setattr( ps, "_enrich_model_info_with_litellm_data", lambda model, **kw: model ) - import litellm.proxy.agent_endpoints.model_list_helpers as mlh - - monkeypatch.setattr( - mlh, - "append_agents_to_model_info", - AsyncMock(side_effect=lambda models, **kw: models), - ) caller = UserAPIKeyAuth( user_id="user-1", @@ -246,13 +232,6 @@ async def test_model_info_v1_restricted_key_filters_deployments(monkeypatch): monkeypatch.setattr( ps, "_enrich_model_info_with_litellm_data", lambda model, **kw: model ) - import litellm.proxy.agent_endpoints.model_list_helpers as mlh - - monkeypatch.setattr( - mlh, - "append_agents_to_model_info", - AsyncMock(side_effect=lambda models, **kw: models), - ) caller = UserAPIKeyAuth( user_id="user-1",