Handles DD API exception (#438)

SneaksAndData · Jun 17, 2024 · 342da3a · 342da3a
1 parent d382e74
commit 342da3a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/adapta/logs/handlers/datadog_api_handler.py b/adapta/logs/handlers/datadog_api_handler.py
@@ -30,7 +30,7 @@
 
 import backoff
 from datadog_api_client import Configuration, ApiClient
-from datadog_api_client.exceptions import ServiceException
+from datadog_api_client.exceptions import ServiceException, ApiException
 from datadog_api_client.v2.api.logs_api import LogsApi
 from datadog_api_client.v2.model.content_encoding import ContentEncoding
 from datadog_api_client.v2.model.http_log import HTTPLog
@@ -167,6 +167,7 @@ def _flush(self) -> None:
                 ConnectionError,
                 HTTPError,
                 ServiceException,
+                ApiException,
             ),
             max_time=self._max_flush_retry_time,
             raise_on_giveup=not self._ignore_flush_failure,