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: fix test to specify allowed_fails

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 21:39:45 -07:00
parent 797a171962
commit ce39649b2a
2 changed files with 17 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/router.py
View file

@ -321,7 +321,10 @@ class Router:
[]
) # initialize an empty list - to allow _add_deployment and delete_deployment to work
self.allowed_fails = allowed_fails or litellm.allowed_fails
if allowed_fails is not None:
self.allowed_fails = allowed_fails
else:
self.allowed_fails = litellm.allowed_fails
self.cooldown_time = cooldown_time or 60
self.disable_cooldowns = disable_cooldowns
self.failed_calls = (