Merge branch 'dev' into d/new-cookbooks

julep-ai · Oct 10, 2024 · dbfde9b · dbfde9b
2 parents a300068 + 4fe49d1
commit dbfde9b
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,
 ]