phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_router.py): handle rate limiting error

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-11 00:00:17 +05:30
parent 61f2fe5837
commit 969594a4b1
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_router.py
View file

@ -495,6 +495,8 @@ def test_img_gen_on_router():
print(response)
assert len(response.data) > 0
router.reset()
except litellm.RateLimitError as e:
pass
except Exception as e:
traceback.print_exc()
pytest.fail(f"Error occurred: {e}")