diff --git a/CHANGELOG.md b/CHANGELOG.md index 12a52a5..7e67846 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,7 @@ +## 0.0.11 + +* **Bugfix: Add UnrecoverableException exception handling back** + ## 0.0.10 * **Bugfix: Add `None` support in mapping `FileDataMeta` response** diff --git a/unstructured_platform_plugins/__version__.py b/unstructured_platform_plugins/__version__.py index cdce53e..f30cf20 100644 --- a/unstructured_platform_plugins/__version__.py +++ b/unstructured_platform_plugins/__version__.py @@ -1 +1 @@ -__version__ = "0.0.10" # pragma: no cover +__version__ = "0.0.11" # pragma: no cover diff --git a/unstructured_platform_plugins/etl_uvicorn/api_generator.py b/unstructured_platform_plugins/etl_uvicorn/api_generator.py index b88d7af..29648b6 100644 --- a/unstructured_platform_plugins/etl_uvicorn/api_generator.py +++ b/unstructured_platform_plugins/etl_uvicorn/api_generator.py @@ -24,6 +24,7 @@ get_schema_dict, map_inputs, ) +from unstructured_platform_plugins.exceptions import UnrecoverableException from unstructured_platform_plugins.schema import FileDataMeta, NewRecord, UsageData from unstructured_platform_plugins.schema.json_schema import ( schema_to_base_model, @@ -158,6 +159,9 @@ async def _stream_response(): status_code=status.HTTP_200_OK, output=output, ) + except UnrecoverableException as ex: + logger.info("Unrecoverable error occurred during plugin invocation") + return InvokeResponse(usage=usage, status_code=512, status_code_text=ex.message) except Exception as invoke_error: logger.error(f"failed to invoke plugin: {invoke_error}", exc_info=True) return InvokeResponse(