fix retry policy

This commit is contained in:
Ishaan Jaff 2024-05-06 16:43:42 -07:00
parent 1de50d62f7
commit 040dc1f23e

View file

@ -3268,6 +3268,8 @@ class Router:
if retry_policy is None: if retry_policy is None:
return None return None
if isinstance(retry_policy, dict):
retry_policy = RetryPolicy(**retry_policy)
if ( if (
isinstance(exception, litellm.BadRequestError) isinstance(exception, litellm.BadRequestError)
and retry_policy.BadRequestErrorRetries is not None and retry_policy.BadRequestErrorRetries is not None