Revert AzureDestination and WorkflowNode changes

awalker4 · awalker4 · commit 1081911f93f2 · 2025-04-17T13:27:04.000-04:00
diff --git a/README.md b/README.md
@@ -126,7 +126,7 @@ with UnstructuredClient() as uc_client:
                 "token": "<value>",
             },
             "name": "<value>",
-            "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+            "type": shared.DestinationConnectorType.AZURE,
         },
     },
         RetryConfig("backoff", BackoffStrategy(1, 50, 1.1, 100), False))
@@ -159,7 +159,7 @@ with UnstructuredClient(
                 "token": "<value>",
             },
             "name": "<value>",
-            "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+            "type": shared.DestinationConnectorType.AZURE,
         },
     })
 
@@ -214,7 +214,7 @@ with UnstructuredClient() as uc_client:
                     "token": "<value>",
                 },
                 "name": "<value>",
-                "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+                "type": shared.DestinationConnectorType.AZURE,
             },
         })
 
@@ -347,7 +347,7 @@ with UnstructuredClient() as uc_client:
                 "token": "<value>",
             },
             "name": "<value>",
-            "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+            "type": shared.DestinationConnectorType.AZURE,
         },
     })
 
@@ -380,7 +380,7 @@ async def main():
                     "token": "<value>",
                 },
                 "name": "<value>",
-                "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+                "type": shared.DestinationConnectorType.AZURE,
             },
         })
 
diff --git a/USAGE.md b/USAGE.md
@@ -17,7 +17,7 @@ with UnstructuredClient() as uc_client:
                 "token": "<value>",
             },
             "name": "<value>",
-            "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+            "type": shared.DestinationConnectorType.AZURE,
         },
     })
 
@@ -50,7 +50,7 @@ async def main():
                     "token": "<value>",
                 },
                 "name": "<value>",
-                "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+                "type": shared.DestinationConnectorType.AZURE,
             },
         })
 
diff --git a/codeSamples.yaml b/codeSamples.yaml
@@ -14,7 +14,7 @@ actions:
       "x-codeSamples":
         - "lang": "python"
           "label": "create_destination"
-          "source": "from unstructured_client import UnstructuredClient\nfrom unstructured_client.models import shared\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.destinations.create_destination(request={\n        \"create_destination_connector\": {\n            \"name\": \"<value>\",\n            \"type\": shared.DestinationConnectorType.AZURE_AI_SEARCH,\n            \"config\": {\n                \"collection_name\": \"<value>\",\n                \"batch_size\": 20,\n                \"api_endpoint\": \"<value>\",\n                \"token\": \"<value>\",\n                \"flatten_metadata\": False,\n            },\n        },\n    })\n\n    assert res.destination_connector_information is not None\n\n    # Handle response\n    print(res.destination_connector_information)"
+          "source": "from unstructured_client import UnstructuredClient\nfrom unstructured_client.models import shared\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.destinations.create_destination(request={\n        \"create_destination_connector\": {\n            \"name\": \"<value>\",\n            \"type\": shared.DestinationConnectorType.ASTRADB,\n            \"config\": {\n                \"remote_url\": \"https://fortunate-appliance.info\",\n                \"anonymous\": False,\n            },\n        },\n    })\n\n    assert res.destination_connector_information is not None\n\n    # Handle response\n    print(res.destination_connector_information)"
   - target: $["paths"]["/api/v1/destinations/{destination_id}"]["delete"]
     update:
       "x-codeSamples":
@@ -33,18 +33,6 @@ actions:
         - "lang": "python"
           "label": "update_destination"
           "source": "from unstructured_client import UnstructuredClient\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.destinations.update_destination(request={\n        \"destination_id\": \"9ea9d27c-4e2e-4cb5-9a69-71bd7ec9d226\",\n        \"update_destination_connector\": {\n            \"config\": {\n                \"host\": \"acidic-patroller.com\",\n                \"database\": \"<value>\",\n                \"port\": 832931,\n                \"username\": \"Salvador.Walsh-Thiel\",\n                \"password\": \"vIMqbllzrdJBExE\",\n                \"table_name\": \"<value>\",\n                \"batch_size\": 977983,\n            },\n        },\n    })\n\n    assert res.destination_connector_information is not None\n\n    # Handle response\n    print(res.destination_connector_information)"
-  - target: $["paths"]["/api/v1/destinations/{destination_id}/connection-check"]["get"]
-    update:
-      "x-codeSamples":
-        - "lang": "python"
-          "label": "get_destination_connection_check_api_v1_destinations__destination_id__connection_check_get"
-          "source": "from unstructured_client import UnstructuredClient\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.destinations.get_destination_connection_check_api_v1_destinations_destination_id_connection_check_get(request={\n        \"destination_id\": \"3c48df35-2b2c-46f2-9aa2-d7eae993797c\",\n    })\n\n    assert res.dag_node_connection_check is not None\n\n    # Handle response\n    print(res.dag_node_connection_check)"
-  - target: $["paths"]["/api/v1/destinations/{destination_id}/connection-check"]["post"]
-    update:
-      "x-codeSamples":
-        - "lang": "python"
-          "label": "check_destination_connection_api_v1_destinations__destination_id__connection_check_post"
-          "source": "from unstructured_client import UnstructuredClient\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.destinations.check_destination_connection_api_v1_destinations_destination_id_connection_check_post(request={\n        \"destination_id\": \"b65169f5-79ba-4464-918f-b0be2c07b962\",\n    })\n\n    assert res.dag_node_connection_check is not None\n\n    # Handle response\n    print(res.dag_node_connection_check)"
   - target: $["paths"]["/api/v1/jobs/"]["get"]
     update:
       "x-codeSamples":
@@ -99,18 +87,6 @@ actions:
         - "lang": "python"
           "label": "update_source"
           "source": "from unstructured_client import UnstructuredClient\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.sources.update_source(request={\n        \"source_id\": \"396edec4-39e8-4ef6-ac69-c9e3ba7a8418\",\n        \"update_source_connector\": {\n            \"config\": {\n                \"host\": \"black-and-white-obesity.info\",\n                \"database\": \"<value>\",\n                \"port\": 848106,\n                \"username\": \"Bruce.Stracke19\",\n                \"password\": \"L8K8WEZq1xvBpSw\",\n                \"table_name\": \"<value>\",\n                \"batch_size\": 177849,\n                \"id_column\": \"id\",\n            },\n        },\n    })\n\n    assert res.source_connector_information is not None\n\n    # Handle response\n    print(res.source_connector_information)"
-  - target: $["paths"]["/api/v1/sources/{source_id}/connection-check"]["get"]
-    update:
-      "x-codeSamples":
-        - "lang": "python"
-          "label": "get_connection_check_api_v1_sources__source_id__connection_check_get"
-          "source": "from unstructured_client import UnstructuredClient\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.sources.get_connection_check_api_v1_sources_source_id_connection_check_get(request={\n        \"source_id\": \"d9c3daec-ddf9-4cef-ae8d-3a989813cc7b\",\n    })\n\n    assert res.dag_node_connection_check is not None\n\n    # Handle response\n    print(res.dag_node_connection_check)"
-  - target: $["paths"]["/api/v1/sources/{source_id}/connection-check"]["post"]
-    update:
-      "x-codeSamples":
-        - "lang": "python"
-          "label": "check_connection_api_v1_sources__source_id__connection_check_post"
-          "source": "from unstructured_client import UnstructuredClient\n\n\nwith UnstructuredClient() as uc_client:\n\n    res = uc_client.sources.check_connection_api_v1_sources_source_id_connection_check_post(request={\n        \"source_id\": \"8cb54f13-5652-423e-889c-1df1f8147126\",\n    })\n\n    assert res.dag_node_connection_check is not None\n\n    # Handle response\n    print(res.dag_node_connection_check)"
   - target: $["paths"]["/api/v1/workflows/"]["get"]
     update:
       "x-codeSamples":
diff --git a/docs/models/shared/azuredestinationconnectorconfig.md b/docs/models/shared/azuredestinationconnectorconfig.md
@@ -0,0 +1,12 @@
+# AzureDestinationConnectorConfig
+
+
+## Fields
+
+| Field                   | Type                    | Required                | Description             |
+| ----------------------- | ----------------------- | ----------------------- | ----------------------- |
+| `account_key`           | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
+| `account_name`          | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
+| `connection_string`     | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
+| `remote_url`            | *str*                   | :heavy_check_mark:      | N/A                     |
+| `sas_token`             | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
diff --git a/docs/models/shared/azuredestinationconnectorconfiginput.md b/docs/models/shared/azuredestinationconnectorconfiginput.md
@@ -0,0 +1,12 @@
+# AzureDestinationConnectorConfigInput
+
+
+## Fields
+
+| Field                   | Type                    | Required                | Description             |
+| ----------------------- | ----------------------- | ----------------------- | ----------------------- |
+| `account_key`           | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
+| `account_name`          | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
+| `connection_string`     | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
+| `remote_url`            | *str*                   | :heavy_check_mark:      | N/A                     |
+| `sas_token`             | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
diff --git a/docs/models/shared/config.md b/docs/models/shared/config.md
@@ -3,6 +3,12 @@
 
 ## Supported Types
 
+### `shared.AzureDestinationConnectorConfigInput`
+
+```python
+value: shared.AzureDestinationConnectorConfigInput = /* values here */
+```
+
 ### `shared.AstraDBConnectorConfigInput`
 
 ```python
diff --git a/docs/models/shared/destinationconnectorinformationconfig.md b/docs/models/shared/destinationconnectorinformationconfig.md
@@ -3,6 +3,12 @@
 
 ## Supported Types
 
+### `shared.AzureDestinationConnectorConfig`
+
+```python
+value: shared.AzureDestinationConnectorConfig = /* values here */
+```
+
 ### `shared.AstraDBConnectorConfig`
 
 ```python
diff --git a/docs/models/shared/destinationconnectortype.md b/docs/models/shared/destinationconnectortype.md
@@ -5,6 +5,7 @@
 
 | Name                             | Value                            |
 | -------------------------------- | -------------------------------- |
+| `AZURE`                          | azure                            |
 | `ASTRADB`                        | astradb                          |
 | `AZURE_AI_SEARCH`                | azure_ai_search                  |
 | `COUCHBASE`                      | couchbase                        |
diff --git a/docs/models/shared/updatedestinationconnectorconfig.md b/docs/models/shared/updatedestinationconnectorconfig.md
@@ -3,6 +3,12 @@
 
 ## Supported Types
 
+### `shared.AzureDestinationConnectorConfigInput`
+
+```python
+value: shared.AzureDestinationConnectorConfigInput = /* values here */
+```
+
 ### `shared.AstraDBConnectorConfigInput`
 
 ```python
diff --git a/docs/models/shared/workflownode.md b/docs/models/shared/workflownode.md
@@ -3,10 +3,10 @@
 
 ## Fields
 
-| Field                   | Type                    | Required                | Description             |
-| ----------------------- | ----------------------- | ----------------------- | ----------------------- |
-| `id`                    | *OptionalNullable[str]* | :heavy_minus_sign:      | N/A                     |
-| `name`                  | *str*                   | :heavy_check_mark:      | N/A                     |
-| `settings`              | Dict[str, *Any*]        | :heavy_minus_sign:      | N/A                     |
-| `subtype`               | *str*                   | :heavy_check_mark:      | N/A                     |
-| `type`                  | *str*                   | :heavy_check_mark:      | N/A                     |
+| Field                                                              | Type                                                               | Required                                                           | Description                                                        |
+| ------------------------------------------------------------------ | ------------------------------------------------------------------ | ------------------------------------------------------------------ | ------------------------------------------------------------------ |
+| `id`                                                               | *OptionalNullable[str]*                                            | :heavy_minus_sign:                                                 | N/A                                                                |
+| `name`                                                             | *str*                                                              | :heavy_check_mark:                                                 | N/A                                                                |
+| `settings`                                                         | Dict[str, *Any*]                                                   | :heavy_minus_sign:                                                 | N/A                                                                |
+| `subtype`                                                          | *str*                                                              | :heavy_check_mark:                                                 | N/A                                                                |
+| `type`                                                             | [shared.WorkflowNodeType](../../models/shared/workflownodetype.md) | :heavy_check_mark:                                                 | N/A                                                                |
diff --git a/docs/models/shared/workflownodetype.md b/docs/models/shared/workflownodetype.md
@@ -0,0 +1,11 @@
+# WorkflowNodeType
+
+
+## Values
+
+| Name        | Value       |
+| ----------- | ----------- |
+| `PARTITION` | partition   |
+| `PROMPTER`  | prompter    |
+| `CHUNK`     | chunk       |
+| `EMBED`     | embed       |
diff --git a/docs/sdks/destinations/README.md b/docs/sdks/destinations/README.md
@@ -34,7 +34,7 @@ with UnstructuredClient() as uc_client:
                 "token": "<value>",
             },
             "name": "<value>",
-            "type": shared.DestinationConnectorType.AZURE_AI_SEARCH,
+            "type": shared.DestinationConnectorType.AZURE,
         },
     })
 
diff --git a/src/unstructured_client/models/shared/__init__.py b/src/unstructured_client/models/shared/__init__.py
@@ -16,6 +16,14 @@
     AzureAISearchConnectorConfigInput,
     AzureAISearchConnectorConfigInputTypedDict,
 )
+from .azuredestinationconnectorconfig import (
+    AzureDestinationConnectorConfig,
+    AzureDestinationConnectorConfigTypedDict,
+)
+from .azuredestinationconnectorconfiginput import (
+    AzureDestinationConnectorConfigInput,
+    AzureDestinationConnectorConfigInputTypedDict,
+)
 from .azuresourceconnectorconfig import (
     AzureSourceConnectorConfig,
     AzureSourceConnectorConfigTypedDict,
@@ -361,6 +369,7 @@
 )
 from .workflowinformation import WorkflowInformation, WorkflowInformationTypedDict
 from .workflownode import WorkflowNode, WorkflowNodeTypedDict
+from .workflownodetype import WorkflowNodeType
 from .workflowschedule import WorkflowSchedule, WorkflowScheduleTypedDict
 from .workflowstate import WorkflowState
 from .workflowtype import WorkflowType
@@ -383,6 +392,10 @@
     "AzureAISearchConnectorConfigInput",
     "AzureAISearchConnectorConfigInputTypedDict",
     "AzureAISearchConnectorConfigTypedDict",
+    "AzureDestinationConnectorConfig",
+    "AzureDestinationConnectorConfigInput",
+    "AzureDestinationConnectorConfigInputTypedDict",
+    "AzureDestinationConnectorConfigTypedDict",
     "AzureSourceConnectorConfig",
     "AzureSourceConnectorConfigInput",
     "AzureSourceConnectorConfigInputTypedDict",
@@ -579,6 +592,7 @@
     "WorkflowInformation",
     "WorkflowInformationTypedDict",
     "WorkflowNode",
+    "WorkflowNodeType",
     "WorkflowNodeTypedDict",
     "WorkflowSchedule",
     "WorkflowScheduleTypedDict",
diff --git a/src/unstructured_client/models/shared/azuredestinationconnectorconfig.py b/src/unstructured_client/models/shared/azuredestinationconnectorconfig.py
@@ -0,0 +1,72 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+from unstructured_client.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+
+
+class AzureDestinationConnectorConfigTypedDict(TypedDict):
+    remote_url: str
+    account_key: NotRequired[Nullable[str]]
+    account_name: NotRequired[Nullable[str]]
+    connection_string: NotRequired[Nullable[str]]
+    sas_token: NotRequired[Nullable[str]]
+
+
+class AzureDestinationConnectorConfig(BaseModel):
+    remote_url: str
+
+    account_key: OptionalNullable[str] = UNSET
+
+    account_name: OptionalNullable[str] = UNSET
+
+    connection_string: OptionalNullable[str] = UNSET
+
+    sas_token: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "account_key",
+            "account_name",
+            "connection_string",
+            "sas_token",
+        ]
+        nullable_fields = [
+            "account_key",
+            "account_name",
+            "connection_string",
+            "sas_token",
+        ]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m
diff --git a/src/unstructured_client/models/shared/azuredestinationconnectorconfiginput.py b/src/unstructured_client/models/shared/azuredestinationconnectorconfiginput.py
diff --git a/src/unstructured_client/models/shared/createdestinationconnector.py b/src/unstructured_client/models/shared/createdestinationconnector.py
diff --git a/src/unstructured_client/models/shared/destinationconnectorinformation.py b/src/unstructured_client/models/shared/destinationconnectorinformation.py
diff --git a/src/unstructured_client/models/shared/destinationconnectortype.py b/src/unstructured_client/models/shared/destinationconnectortype.py
diff --git a/src/unstructured_client/models/shared/updatedestinationconnector.py b/src/unstructured_client/models/shared/updatedestinationconnector.py
diff --git a/src/unstructured_client/models/shared/workflownode.py b/src/unstructured_client/models/shared/workflownode.py
diff --git a/src/unstructured_client/models/shared/workflownodetype.py b/src/unstructured_client/models/shared/workflownodetype.py
