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(test_router_fallbacks.py): fix tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-30 18:48:39 -07:00
parent a12878b0f8
commit b4ec16fd5f
2 changed files with 24 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

25
litellm/router.py
View file

@ -1512,31 +1512,6 @@ class Router:
## LOGGING
kwargs = self.log_retry(kwargs=kwargs, e=e)
remaining_retries = num_retries - current_attempt
# if "No models available" in str(e):
# timeout = litellm._calculate_retry_after(
# remaining_retries=remaining_retries,
# max_retries=num_retries,
# min_timeout=self.retry_after,
# )
# await asyncio.sleep(timeout)
# elif (
# hasattr(e, "status_code")
# and hasattr(e, "response")
# and litellm._should_retry(status_code=e.status_code)
# ):
# if hasattr(e.response, "headers"):
# timeout = litellm._calculate_retry_after(
# remaining_retries=remaining_retries,
# max_retries=num_retries,
# response_headers=e.response.headers,
# min_timeout=self.retry_after,
# )
# else:
# timeout = litellm._calculate_retry_after(
# remaining_retries=remaining_retries,
# max_retries=num_retries,
# min_timeout=self.retry_after,
# )
_timeout = self._router_should_retry(
e=original_exception,
remaining_retries=remaining_retries,