diff --git a/litellm/router.py b/litellm/router.py index 9c56445d32b..49699ffed22 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -87,7 +87,7 @@ class Router: RetryPolicy ] = None, # set custom retries for different exceptions model_group_retry_policy: Optional[ - dict[str, RetryPolicy] + Dict[str, RetryPolicy] ] = {}, # set custom retry policies based on model group allowed_fails: Optional[ int