Add litellm exceptions to the list of non-retryable errors

julep-ai · Oct 9, 2024 · 018360c · 018360c
1 parent 7cfa193
commit 018360c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/agents-api/agents_api/common/exceptions/tasks.py b/agents-api/agents_api/common/exceptions/tasks.py
@@ -14,6 +14,7 @@
 import httpx
 import jinja2
 import jsonschema.exceptions
+import litellm
 import pydantic
 import requests
 import temporalio.exceptions
@@ -70,10 +71,12 @@
     pydantic.ValidationError,
     requests.exceptions.InvalidURL,
     requests.exceptions.MissingSchema,
+    #
     # Box exceptions
     box.exceptions.BoxKeyError,
     box.exceptions.BoxTypeError,
     box.exceptions.BoxValueError,
+    #
     # Beartype exceptions
     beartype.roar.BeartypeException,
     beartype.roar.BeartypeDecorException,
@@ -88,6 +91,14 @@
     beartype.roar.BeartypeCallHintReturnViolation,
     beartype.roar.BeartypeDecorHintParamDefaultViolation,
     beartype.roar.BeartypeDoorHintViolation,
+    #
+    # LiteLLM exceptions
+    litellm.exceptions.NotFoundError,
+    litellm.exceptions.InvalidRequestError,
+    litellm.exceptions.AuthenticationError,
+    litellm.exceptions.ServiceUnavailableError,
+    litellm.exceptions.OpenAIError,
+    litellm.exceptions.APIError,
 ]