diff --git a/litellm/proxy/management_endpoints/model_management_endpoints.py b/litellm/proxy/management_endpoints/model_management_endpoints.py index ab5e2697bb0..d6d7d9ea03a 100644 --- a/litellm/proxy/management_endpoints/model_management_endpoints.py +++ b/litellm/proxy/management_endpoints/model_management_endpoints.py @@ -1715,11 +1715,9 @@ async def clear_cache(): for model_id in db_model_ids: llm_router.delete_deployment(id=model_id) - # Clear all strategy routers so they are re-initialized from DB state llm_router.auto_routers.clear() llm_router.complexity_routers.clear() llm_router.quality_routers.clear() - llm_router.adaptive_routers.clear() # Reload only DB models await proxy_config.add_deployment(prisma_client=prisma_client, proxy_logging_obj=proxy_logging_obj) diff --git a/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py index a7a7f58ffd1..f915024503c 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py @@ -463,8 +463,6 @@ class TestClearCache: mock_router.complexity_routers.clear = MagicMock() mock_router.quality_routers = MagicMock() mock_router.quality_routers.clear = MagicMock() - mock_router.adaptive_routers = MagicMock() - mock_router.adaptive_routers.clear = MagicMock() mock_config = MagicMock() mock_config.add_deployment = AsyncMock(return_value=True) @@ -486,11 +484,9 @@ class TestClearCache: mock_router.delete_deployment.assert_any_call(id="db-model-1") mock_router.delete_deployment.assert_any_call(id="db-model-2") - # Should have cleared all strategy routers mock_router.auto_routers.clear.assert_called_once() mock_router.complexity_routers.clear.assert_called_once() mock_router.quality_routers.clear.assert_called_once() - mock_router.adaptive_routers.clear.assert_called_once() # Should have called add_deployment to reload DB models mock_config.add_deployment.assert_called_once_with(