add litellm error to the default retry exceptions

mayinghan · mayinghan · commit 77265afaa0fc · 2025-10-24T17:10:26.000-07:00
diff --git a/eval_protocol/pytest/exception_config.py b/eval_protocol/pytest/exception_config.py
@@ -8,7 +8,7 @@
 
 import backoff
 
-
+import litellm
 import requests
 import httpx
 
@@ -28,8 +28,14 @@
     httpx.TimeoutException,
     httpx.NetworkError,
     httpx.RemoteProtocolError,
-}
 
+    litellm.exceptions.RateLimitError,
+    litellm.exceptions.InternalServerError,
+    litellm.exceptions.Timeout,
+    litellm.exceptions.NotFoundError,
+    litellm.exceptions.BadRequestError,
+    litellm.exceptions.ServiceUnavailableError,
+}
 
 @dataclass
 class BackoffConfig:
