diff --git a/litellm/router.py b/litellm/router.py index cd6e544eb33..51f532ca3c4 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -9735,7 +9735,6 @@ class Router: "retry_after", "fallbacks", "context_window_fallbacks", - "retry_policy", "model_group_retry_policy", "retry_policy", "model_group_alias", diff --git a/tests/router_unit_tests/test_router_helper_utils.py b/tests/router_unit_tests/test_router_helper_utils.py index d5fe819008e..5b62e80a129 100644 --- a/tests/router_unit_tests/test_router_helper_utils.py +++ b/tests/router_unit_tests/test_router_helper_utils.py @@ -2838,9 +2838,10 @@ def test_upsert_deployment_clears_stale_budget_config(monkeypatch): def test_update_router_config_accepts_retry_policy(): """GH#31308: UpdateRouterConfig must accept retry_policy so it is not silently dropped when set via UI or /config/update.""" - from litellm.types.router import UpdateRouterConfig + from litellm.types.router import UpdateRouterConfig, RetryPolicy config = UpdateRouterConfig( retry_policy={"RateLimitErrorRetries": 3} ) - assert config.retry_policy == {"RateLimitErrorRetries": 3} + assert isinstance(config.retry_policy, RetryPolicy) + assert config.retry_policy.RateLimitErrorRetries == 3