phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(router.py): fix max retries on set_client

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-24 22:02:48 -07:00
parent 242830108c
commit 821844c1a3
2 changed files with 18 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_router.py
View file

@ -18,6 +18,12 @@ from dotenv import load_dotenv
load_dotenv()
def test_router_num_retries_init():
router = Router(num_retries=0)
assert router.num_retries == 0
def test_exception_raising():
# this tests if the router raises an exception when invalid params are set
# in this test both deployments have bad keys - Keep this test. It validates if the router raises the most recent exception