unstructured-platform-plugins 0.0.0__tar.gz → 0.0.2__tar.gz

{unstructured-platform-plugins-0.0.0/unstructured_platform_plugins.egg-info → unstructured-platform-plugins-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unstructured-platform-plugins
-Version: 0.0.0
+Version: 0.0.2
 Summary: A utility library that provides tools around unstructured plugin development
 Home-page: https://github.com/Unstructured-IO/unstructured-platform-plugins
 License: Apache-2.0

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2}/test/test_schema.py

@@ -7,7 +7,7 @@ from typing import Any, Optional, Union
 import pytest
 from pydantic import BaseModel
 from typing_extensions import TypedDict
-from unstructured.ingest.v2.interfaces import FileData
+from unstructured_ingest.v2.interfaces import FileData
 
 import unstructured_platform_plugins.schema.json_schema as js
 from unstructured_platform_plugins.etl_uvicorn.utils import get_input_schema
@@ -507,7 +507,7 @@ def test_file_data():
                         ],
                         "default": None,
                     },
-                    "doc_type": {"type": "string", "enum": ["batch", "file"], "default": "file"},
+                    "doc_type": {"type": "string", "default": "file"},
                     "metadata": {
                         "type": "object",
                         "properties": {
@@ -599,7 +599,7 @@ def test_file_data():
                     ],
                     "default": None,
                 },
-                "doc_type": {"type": "string", "enum": ["batch", "file"], "default": "file"},
+                "doc_type": {"type": "string", "default": "file"},
                 "metadata": {
                     "type": "object",
                     "properties": {

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2}/test/test_utils.py

@@ -4,7 +4,7 @@ from enum import Enum
 
 import pytest
 from pydantic import BaseModel
-from unstructured.ingest.v2.interfaces import FileData
+from unstructured_ingest.v2.interfaces import FileData
 from uvicorn.importer import import_from_string
 
 from unstructured_platform_plugins.etl_uvicorn import utils

unstructured-platform-plugins-0.0.2/unstructured_platform_plugins/__version__.py

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

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2}/unstructured_platform_plugins/etl_uvicorn/api_generator.py

@@ -122,8 +122,8 @@ def generate_fast_api(
 
     else:
 
-        @fastapi_app.post("/invoke", response_model=response_type)
-        async def run_job() -> response_type:
+        @fastapi_app.post("/invoke", response_model=InvokeResponse)
+        async def run_job() -> InvokeResponse:
             logger.debug(f"invoking function without inputs: {func}")
             return await wrap_fn(
                 func=func,

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2}/unstructured_platform_plugins/etl_uvicorn/utils.py

@@ -1,6 +1,7 @@
 import inspect
 from dataclasses import is_dataclass
 from enum import EnumMeta
+from inspect import Parameter
 from types import GenericAlias, NoneType
 from typing import Any, Callable, Optional
 
@@ -58,6 +59,10 @@ def get_input_schema(func: Callable, omit: Optional[list[str]] = None) -> dict:
     parameters = get_typed_parameters(func)
     if omit:
         parameters = [p for p in parameters if p.name not in omit]
+    # Omit self if wrapping method
+    parameters = [
+        p for p in parameters if not (p.param_type is Parameter.empty and p.name == "self")
+    ]
     return parameters_to_json_schema(parameters)
 
 

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2}/unstructured_platform_plugins/schema/json_schema.py

@@ -4,11 +4,11 @@ from enum import Enum, EnumMeta
 from inspect import Parameter
 from pathlib import Path
 from types import GenericAlias, NoneType, UnionType
-from typing import Any, Optional, Type, Union, _UnionGenericAlias
+from typing import Any, Literal, Optional, Type, Union, _UnionGenericAlias
 
 from pydantic import BaseModel, create_model
 from pydantic.fields import FieldInfo, PydanticUndefined
-from unstructured.ingest.v2.interfaces import FileData
+from unstructured_ingest.v2.interfaces import FileData
 
 from unstructured_platform_plugins.schema.utils import TypedParameter
 from unstructured_platform_plugins.type_hints import get_type_hints
@@ -37,6 +37,13 @@ def is_typed_dict(val: Any) -> bool:
 
 
 def type_to_json_schema(t: Type, args: Optional[tuple[Any, ...]] = None) -> dict:
+    if t is Literal:
+        arg_types = [type(a) for a in args]
+        if len(set(arg_types)) != 1:
+            raise TypeError(
+                "Literal must be of a single type. Defined with: {}".format(", ".join(arg_types))
+            )
+        t = arg_types[0]
     resp = {"type": types_map[t]}
     if t is list and args:
         list_type = args[0]
@@ -305,6 +312,8 @@ def schema_to_base_model_type(json_type_name, name: str, type_info: dict) -> Typ
 
 def schema_to_base_model(schema: dict, name: str = "reconstructed_model") -> Type[BaseModel]:
     inputs = {}
+    if schema["type"] == "null":
+        return create_model(name)
     properties = schema["properties"]
 
     for k, v in properties.items():

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2/unstructured_platform_plugins.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unstructured-platform-plugins
-Version: 0.0.0
+Version: 0.0.2
 Summary: A utility library that provides tools around unstructured plugin development
 Home-page: https://github.com/Unstructured-IO/unstructured-platform-plugins
 License: Apache-2.0

{unstructured-platform-plugins-0.0.0 → unstructured-platform-plugins-0.0.2}/unstructured_platform_plugins.egg-info/requires.txt

@@ -1,6 +1,6 @@
-fastapi
 click
-unstructured
 uvicorn
-requests
+unstructured-ingest
+fastapi
 click
+requests

unstructured-platform-plugins-0.0.0/unstructured_platform_plugins/__version__.py

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