diff --git a/unstructured_platform_plugins/__version__.py b/unstructured_platform_plugins/__version__.py index ff3da46..16618bc 100644 --- a/unstructured_platform_plugins/__version__.py +++ b/unstructured_platform_plugins/__version__.py @@ -1 +1 @@ -__version__ = "0.0.33" # pragma: no cover +__version__ = "0.0.34" # pragma: no cover diff --git a/unstructured_platform_plugins/etl_uvicorn/api_generator.py b/unstructured_platform_plugins/etl_uvicorn/api_generator.py index df40173..6c37db4 100644 --- a/unstructured_platform_plugins/etl_uvicorn/api_generator.py +++ b/unstructured_platform_plugins/etl_uvicorn/api_generator.py @@ -216,8 +216,8 @@ 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_text=f"[{exc.__class__.__name__}] {exc}", + status_code=exc.status_code, + status_code_text=exc.detail, file_data=request_dict.get("file_data", None), ) except UnrecoverableException as ex: diff --git a/unstructured_platform_plugins/etl_uvicorn/errors.py b/unstructured_platform_plugins/etl_uvicorn/errors.py index f6ca371..8530243 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 @@ -36,24 +32,18 @@ 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): + """ + Wraps an exception in a HTTPException. + """ + 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):