From da6e63112e112c7deb4a6e0b8399e0efd52b4adc Mon Sep 17 00:00:00 2001 From: Nathan Van Gheem Date: Mon, 25 Nov 2024 13:39:43 -0500 Subject: [PATCH 1/3] Add UnrecoverableException exception handling back --- CHANGELOG.md | 4 ++++ unstructured_platform_plugins/etl_uvicorn/api_generator.py | 4 ++++ 2 files changed, 8 insertions(+) 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/etl_uvicorn/api_generator.py b/unstructured_platform_plugins/etl_uvicorn/api_generator.py index b88d7af..4bdd509 100644 --- a/unstructured_platform_plugins/etl_uvicorn/api_generator.py +++ b/unstructured_platform_plugins/etl_uvicorn/api_generator.py @@ -14,6 +14,7 @@ from unstructured_ingest.v2.interfaces import FileData from uvicorn.config import LOG_LEVELS from uvicorn.importer import import_from_string +from unstructured_platform_plugins.exceptions import UnrecoverableException from unstructured_platform_plugins.etl_uvicorn.otel import get_metric_provider, get_trace_provider from unstructured_platform_plugins.etl_uvicorn.utils import ( @@ -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( From 2063ea78cdc625054a4a8573df762818761b36cc Mon Sep 17 00:00:00 2001 From: Nathan Van Gheem Date: Mon, 25 Nov 2024 13:39:59 -0500 Subject: [PATCH 2/3] . --- unstructured_platform_plugins/etl_uvicorn/api_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unstructured_platform_plugins/etl_uvicorn/api_generator.py b/unstructured_platform_plugins/etl_uvicorn/api_generator.py index 4bdd509..29648b6 100644 --- a/unstructured_platform_plugins/etl_uvicorn/api_generator.py +++ b/unstructured_platform_plugins/etl_uvicorn/api_generator.py @@ -14,7 +14,6 @@ from unstructured_ingest.v2.interfaces import FileData from uvicorn.config import LOG_LEVELS from uvicorn.importer import import_from_string -from unstructured_platform_plugins.exceptions import UnrecoverableException from unstructured_platform_plugins.etl_uvicorn.otel import get_metric_provider, get_trace_provider from unstructured_platform_plugins.etl_uvicorn.utils import ( @@ -25,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, From 824e75247ee43cf095a21dd82b5c951c5e008c7f Mon Sep 17 00:00:00 2001 From: Nathan Van Gheem Date: Mon, 25 Nov 2024 13:40:55 -0500 Subject: [PATCH 3/3] bump --- unstructured_platform_plugins/__version__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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