From 3d8adc9925dc2127b69da1c280d479e617df2eeb Mon Sep 17 00:00:00 2001 From: 72004 Date: Fri, 21 Aug 2026 19:36:20 +0500 Subject: [PATCH] test(router): cover internal server retry helper directly --- .../test_router_helper_utils.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/router_unit_tests/test_router_helper_utils.py b/tests/router_unit_tests/test_router_helper_utils.py index 62ccfa44f66..108c4578027 100644 --- a/tests/router_unit_tests/test_router_helper_utils.py +++ b/tests/router_unit_tests/test_router_helper_utils.py @@ -1382,6 +1382,24 @@ def test_get_num_retries_from_retry_policy( assert calc_num_retries == num_retries +def test_get_num_retries_from_retry_policy_handles_internal_server_error_directly(): + from litellm.router_utils.get_retry_from_policy import ( + get_num_retries_from_retry_policy, + ) + from litellm.types.router import RetryPolicy + + calc_num_retries = get_num_retries_from_retry_policy( + exception=litellm.InternalServerError( + message="test", + llm_provider="openai", + model="gpt-5-mini", + ), + retry_policy=RetryPolicy(InternalServerErrorRetries=5), + ) + + assert calc_num_retries == 5 + + @pytest.mark.parametrize( "exception_type, exception_name, allowed_fails", [