unstructured-platform-plugins 0.0.28__tar.gz → 0.0.30__tar.gz

{unstructured_platform_plugins-0.0.28 → unstructured_platform_plugins-0.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unstructured_platform_plugins
-Version: 0.0.28
+Version: 0.0.30
 Summary: Wrapper to convert arbitrary code into a uvicorn/fastapi implementation for Unstructured Platform
 License-File: LICENSE.md
 Classifier: Development Status :: 4 - Beta
@@ -17,6 +17,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.10
 Requires-Dist: click
+Requires-Dist: dataclasses-json
 Requires-Dist: fastapi
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc
 Requires-Dist: opentelemetry-instrumentation-fastapi

{unstructured_platform_plugins-0.0.28 → unstructured_platform_plugins-0.0.30}/pyproject.toml

@@ -26,6 +26,7 @@ dependencies = [
     "unstructured-ingest",
     "opentelemetry-instrumentation-fastapi",
     "opentelemetry-exporter-otlp-proto-grpc",
+    "dataclasses-json"
 ]
 dynamic = [
  "version",

unstructured_platform_plugins-0.0.30/unstructured_platform_plugins/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.0.30"  # pragma: no cover

{unstructured_platform_plugins-0.0.28 → unstructured_platform_plugins-0.0.30}/unstructured_platform_plugins/etl_uvicorn/api_generator.py

@@ -11,7 +11,7 @@ from fastapi.responses import StreamingResponse
 from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
 from pydantic import BaseModel, Field, create_model
 from starlette.responses import RedirectResponse
-from unstructured_ingest.data_types.file_data import file_data_from_dict
+from unstructured_ingest.data_types.file_data import BatchFileData, FileData, file_data_from_dict
 from uvicorn.config import LOG_LEVELS
 from uvicorn.importer import import_from_string
 
@@ -31,6 +31,8 @@ from unstructured_platform_plugins.schema.json_schema import (
     schema_to_base_model,
 )
 
+FileDataType = Union[FileData, BatchFileData]
+
 
 class EtlApiException(Exception):
     pass
@@ -137,7 +139,8 @@ def _wrap_in_fastapi(
     class InvokeResponse(BaseModel):
         usage: list[UsageData]
         status_code: int
-        filedata_meta: Optional[filedata_meta_model]
+        file_data: Optional[FileDataType] = None
+        filedata_meta: Optional[filedata_meta_model] = None
         status_code_text: Optional[str] = None
         output: Optional[response_type] = None
         message_channels: MessageChannels = Field(default_factory=MessageChannels)
@@ -177,6 +180,7 @@ def _wrap_in_fastapi(
                                     ),
                                     status_code=status.HTTP_200_OK,
                                     output=output,
+                                    file_data=request_dict.get("file_data", None),
                                 ).model_dump_json()
                                 + "\n"
                             )
@@ -202,6 +206,7 @@ def _wrap_in_fastapi(
                     filedata_meta=filedata_meta_model.model_validate(filedata_meta.model_dump()),
                     status_code=status.HTTP_200_OK,
                     output=output,
+                    file_data=request_dict.get("file_data", None),
                 )
         except UnrecoverableException as ex:
             logger.info("Unrecoverable error occurred during plugin invocation")
@@ -211,6 +216,7 @@ def _wrap_in_fastapi(
                 status_code=512,
                 status_code_text=ex.message,
                 filedata_meta=filedata_meta_model.model_validate(filedata_meta.model_dump()),
+                file_data=request_dict.get("file_data", None),
             )
         except Exception as invoke_error:
             logger.error(f"failed to invoke plugin: {invoke_error}", exc_info=True)
@@ -221,6 +227,7 @@ def _wrap_in_fastapi(
                 filedata_meta=filedata_meta_model.model_validate(filedata_meta.model_dump()),
                 status_code=http_error.status_code,
                 status_code_text=f"[{invoke_error.__class__.__name__}] {invoke_error}",
+                file_data=request_dict.get("file_data", None),
             )
 
     if input_schema_model.model_fields:

unstructured_platform_plugins-0.0.28/unstructured_platform_plugins/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.0.28"  # pragma: no cover