From 9d4084406a24cde9503b4741da9aee7641d7d8ed Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Mon, 30 Jun 2025 08:52:23 -0700 Subject: [PATCH 1/3] temp --- unstructured_platform_plugins/__version__.py | 2 +- .../etl_uvicorn/api_generator.py | 4 ++-- .../etl_uvicorn/errors.py | 16 +++------------- 3 files changed, 6 insertions(+), 16 deletions(-) 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..cf1372b 100644 --- a/unstructured_platform_plugins/etl_uvicorn/errors.py +++ b/unstructured_platform_plugins/etl_uvicorn/errors.py @@ -20,9 +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 +33,17 @@ 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: + """ + Wraps an exception in a HTTPException. + """ if 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): From 54109972a3f3f2241da7d598f6cc1aba24a75b4c Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Mon, 30 Jun 2025 08:52:41 -0700 Subject: [PATCH 2/3] lint --- unstructured_platform_plugins/etl_uvicorn/errors.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/unstructured_platform_plugins/etl_uvicorn/errors.py b/unstructured_platform_plugins/etl_uvicorn/errors.py index cf1372b..f255ef1 100644 --- a/unstructured_platform_plugins/etl_uvicorn/errors.py +++ b/unstructured_platform_plugins/etl_uvicorn/errors.py @@ -20,7 +20,6 @@ class UserAuthError(UserError): status_code: int = 403 - class RateLimitError(UserError): status_code: int = 429 @@ -33,7 +32,6 @@ class ProviderError(BaseError): status_code: int = 500 - class CatchAllError(BaseError): status_code: int = 512 From e49572c0341fcf7dc389f13d19215099c26ab5a8 Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Mon, 30 Jun 2025 09:11:18 -0700 Subject: [PATCH 3/3] safety --- unstructured_platform_plugins/etl_uvicorn/errors.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/unstructured_platform_plugins/etl_uvicorn/errors.py b/unstructured_platform_plugins/etl_uvicorn/errors.py index f255ef1..8530243 100644 --- a/unstructured_platform_plugins/etl_uvicorn/errors.py +++ b/unstructured_platform_plugins/etl_uvicorn/errors.py @@ -40,7 +40,9 @@ def wrap_error(e: Exception) -> HTTPException: """ Wraps an exception in a 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, ingest_errors.RateLimitError): return RateLimitError(e)