diff --git a/unstructured_platform_plugins/etl_uvicorn/api_generator.py b/unstructured_platform_plugins/etl_uvicorn/api_generator.py index df40173..d08369d 100644 --- a/unstructured_platform_plugins/etl_uvicorn/api_generator.py +++ b/unstructured_platform_plugins/etl_uvicorn/api_generator.py @@ -216,7 +216,7 @@ async def _stream_response(): usage=usage, message_channels=message_channels, filedata_meta=filedata_meta_model.model_validate(filedata_meta.model_dump()), - status_code=wrap_error(exc).status_code, + status_code=exc.status_code, status_code_text=f"[{exc.__class__.__name__}] {exc}", file_data=request_dict.get("file_data", None), ) diff --git a/unstructured_platform_plugins/etl_uvicorn/errors.py b/unstructured_platform_plugins/etl_uvicorn/errors.py index f6ca371..ec32de9 100644 --- a/unstructured_platform_plugins/etl_uvicorn/errors.py +++ b/unstructured_platform_plugins/etl_uvicorn/errors.py @@ -20,10 +20,6 @@ class UserAuthError(UserError): status_code: int = 403 -class UnprocessableEntityError(UserError): - status_code: int = 422 - - class RateLimitError(UserError): status_code: int = 429 @@ -45,15 +41,10 @@ class CatchAllError(BaseError): def wrap_error(e: Exception) -> HTTPException: - if isinstance(e, ingest_errors.UserAuthError): + if isinstance(e, HTTPException): + return e + elif isinstance(e, ingest_errors.UserAuthError): return UserAuthError(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):