From 040dc1f23e7a29147a18221e66f18eaf0c58b49c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 6 May 2024 16:43:42 -0700 Subject: [PATCH] fix retry policy --- litellm/router.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/router.py b/litellm/router.py index 303fab128..9fff3015c 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -3268,6 +3268,8 @@ class Router: if retry_policy is None: return None + if isinstance(retry_policy, dict): + retry_policy = RetryPolicy(**retry_policy) if ( isinstance(exception, litellm.BadRequestError) and retry_policy.BadRequestErrorRetries is not None