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

fix - return in LITELLM_EXCEPTION_TYPES

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-01 17:05:33 -07:00
parent 5236701d21
commit a485b19215
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/router.py
View file

@ -2295,6 +2295,10 @@ class Router:
healthy_deployments=_healthy_deployments, healthy_deployments=_healthy_deployments,
) )
time.sleep(_timeout) time.sleep(_timeout)
if type(original_exception) in litellm.LITELLM_EXCEPTION_TYPES:
original_exception.max_retries = num_retries
original_exception.num_retries = current_attempt
raise original_exception raise original_exception
### HELPER FUNCTIONS ### HELPER FUNCTIONS