scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix router retries tests

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-20 13:02:24 -07:00
parent c4153f9340
commit cae9245b33
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_router_retries.py
View file

@ -392,7 +392,7 @@ def test_retry_rate_limit_error_with_healthy_deployments():
def test_do_retry_rate_limit_error_with_no_fallbacks_and_no_healthy_deployments():
"""
Test 2. It SHOULD Retry, when healthy_deployments is [] and fallbacks is None
Test 2. It SHOULD NOT Retry, when healthy_deployments is [] and fallbacks is None
"""
healthy_deployments = []
@ -415,9 +415,10 @@ def test_do_retry_rate_limit_error_with_no_fallbacks_and_no_healthy_deployments(
response = router.should_retry_this_error(
error=rate_limit_error, healthy_deployments=healthy_deployments
)
assert response == True
pytest.fail("Should have raised an error")
except Exception as e:
pytest.fail("Should not have failed this error - {}".format(str(e)))
print("got an exception", e)
pass
def test_raise_context_window_exceeded_error():