scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

test(router): cover internal server retry helper directly

Browse source
This commit is contained in:
72004 2026-08-21 19:36:20 +05:00
parent f2d40563d4
commit 3d8adc9925
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/router_unit_tests/test_router_helper_utils.py
View file

@ -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",
[