Revert our custom patch changes

Unstructured-IO · Sep 19, 2024 · 830bee6 · 830bee6
1 parent 4d716d0
commit 830bee6
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 41 deletions.
diff --git a/Makefile b/Makefile
@@ -68,10 +68,6 @@ client-generate-local:
 	speakeasy overlay apply -s ./openapi.json -o ./overlay_client.yaml > ./openapi_client.json
 	speakeasy generate sdk -s ./openapi_client.json -o ./ -l python
 
-.PHONY: patch-custom-code
-patch-custom-code:
-	git apply _custom_code.patch
-
 .PHONY: publish
 publish:
 	./scripts/publish.sh

diff --git a/_custom_code.patch b/_custom_code.patch
diff --git a/_test_unstructured_client/integration/test_decorators.py b/_test_unstructured_client/integration/test_decorators.py
@@ -356,7 +356,7 @@ async def mock_send(_, request: httpx.Request, **kwargs):
         )
 
     req = operations.PartitionRequest(
-        shared.PartitionParameters(
+        partition_parameters=shared.PartitionParameters(
             files=files,
             split_pdf_page=True,
             split_pdf_allow_failed=False,

diff --git a/_test_unstructured_client/unit/test_custom_hooks.py b/_test_unstructured_client/unit/test_custom_hooks.py
@@ -122,11 +122,11 @@ def mock_post(request):
         files = shared.Files(content=f.read(), file_name=filename)
 
     req = operations.PartitionRequest(
-        shared.PartitionParameters(files=files)
+        partition_parameters=shared.PartitionParameters(files=files)
     )
 
     with pytest.raises(Exception, match=f"Status {status_code}"):
-        session.general.partition(req, retries=retries)
+        session.general.partition(request=req, retries=retries)
 
     if expect_retry:
         assert number_of_requests[0] > 1

diff --git a/src/unstructured_client/general.py b/src/unstructured_client/general.py
@@ -11,10 +11,10 @@
 class General(BaseSDK):
     def partition(
         self,
+        *,
         request: Union[
             operations.PartitionRequest, operations.PartitionRequestTypedDict
         ],
-        *,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -116,10 +116,10 @@ def partition(
 
     async def partition_async(
         self,
+        *,
         request: Union[
             operations.PartitionRequest, operations.PartitionRequestTypedDict
         ],
-        *,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,

diff --git a/src/unstructured_client/models/operations/partition.py b/src/unstructured_client/models/operations/partition.py
@@ -25,9 +25,6 @@ class PartitionRequestTypedDict(TypedDict):
 
 
 class PartitionRequest(BaseModel):
-    def __init__(self, partition_parameters, **kwargs):
-        BaseModel.__init__(self, partition_parameters=partition_parameters, **kwargs)
-
     partition_parameters: Annotated[
         shared_partition_parameters.PartitionParameters,
         FieldMetadata(request=RequestMetadata(media_type="multipart/form-data")),