Handle 400, 422 and 504 HTTPExceptions

Author: PastelStorm

unstructured_platform_plugins/etl_uvicorn/errors.py

@@ -36,15 +36,24 @@ class ProviderError(BaseError):
     status_code: int = 500
 
 
+class GatewayTimeoutError(BaseError):
+    status_code: int = 504
+
+
 class CatchAllError(BaseError):
     status_code: int = 512
 
 
 def wrap_error(e: Exception) -> HTTPException:
     if isinstance(e, ingest_errors.UserAuthError):
         return UserAuthError(e)
-    elif isinstance(e, ingest_errors.UnprocessableEntityError):
-        return UnprocessableEntityError(e)
+    elif isinstance(e, HTTPException):
+        if e.status_code == 400:
+            return UserError(e)
+        if e.status_code == 422:
+            return UnprocessableEntityError(e)
+        if e.status_code == 504:
+            return GatewayTimeoutError(e)
     elif isinstance(e, ingest_errors.RateLimitError):
         return RateLimitError(e)
     elif isinstance(e, ingest_errors.QuotaError):