Just patch general.py and ignore it

Author: awalker4

.genignore

@@ -6,4 +6,12 @@ _sample_docs
 _test_unstructured_client
 
 # ignore Makefile
-Makefile
+Makefile
+
+# Ignore general.py so we can patch in our partitioning url
+# If we ever have a new endpoint under /general, we need to:
+#  - Comment out this ignore line
+#  - Generate locally, watch the new endpoint appear
+#  - Adjust the custom url snippets in the file
+#  - Bring back the ignore line and commit
+src/unstructured_client/general.py

src/unstructured_client/general.py

@@ -48,6 +48,12 @@ def partition(
         else:
             base_url = self._get_url(base_url, url_variables)
 
+        # Note(austin): Add a custom check to handle the default server URL
+        # The SDK globally defaults to the platform URL.
+        # If that hasn't changed, we need to switch to the partition url here.
+        if base_url == "https://platform.unstructuredapp.io":
+            base_url = "https://api.unstructuredapp.io"
+
         if not isinstance(request, BaseModel):
             request = utils.unmarshal(request, operations.PartitionRequest)
         request = cast(operations.PartitionRequest, request)
@@ -175,6 +181,12 @@ async def partition_async(
         else:
             base_url = self._get_url(base_url, url_variables)
 
+        # Note(austin): Add a custom check to handle the default server URL
+        # The SDK globally defaults to the platform URL.
+        # If that hasn't changed, we need to switch to the partition url here.
+        if base_url == "https://platform.unstructuredapp.io":
+            base_url = "https://api.unstructuredapp.io"
+
         if not isinstance(request, BaseModel):
             request = utils.unmarshal(request, operations.PartitionRequest)
         request = cast(operations.PartitionRequest, request)