databricks-sdk 0.20.0__py3-none-any.whl → 0.21.0__py3-none-any.whl

databricks/sdk/service/pipelines.py

@@ -8,7 +8,7 @@ import time
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
-from typing import Any, Callable, Dict, Iterator, List, Optional
+from typing import Callable, Dict, Iterator, List, Optional
 
 from ..errors import OperationFailed
 from ._internal import Wait, _enum, _from_dict, _repeated_dict
@@ -178,14 +178,14 @@ class DataPlaneId:
     instance: Optional[str] = None
     """The instance name of the data plane emitting an event."""
 
-    seq_no: Optional[Any] = None
+    seq_no: Optional[int] = None
     """A sequence number, unique and increasing within the data plane instance."""
 
     def as_dict(self) -> dict:
         """Serializes the DataPlaneId into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.instance is not None: body['instance'] = self.instance
-        if self.seq_no: body['seq_no'] = self.seq_no
+        if self.seq_no is not None: body['seq_no'] = self.seq_no
         return body
 
     @classmethod
@@ -194,6 +194,20 @@ class DataPlaneId:
         return cls(instance=d.get('instance', None), seq_no=d.get('seq_no', None))
 
 
+@dataclass
+class DeletePipelineResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeletePipelineResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeletePipelineResponse:
+        """Deserializes the DeletePipelineResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EditPipeline:
     allow_duplicate_names: Optional[bool] = None
@@ -313,6 +327,20 @@ class EditPipeline:
                    trigger=_from_dict(d, 'trigger', PipelineTrigger))
 
 
+@dataclass
+class EditPipelineResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditPipelineResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditPipelineResponse:
+        """Deserializes the EditPipelineResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ErrorDetail:
     exceptions: Optional[List[SerializedException]] = None
@@ -566,6 +594,20 @@ class ListUpdatesResponse:
                    updates=_repeated_dict(d, 'updates', UpdateInfo))
 
 
+@dataclass
+class ManualTrigger:
+
+    def as_dict(self) -> dict:
+        """Serializes the ManualTrigger into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ManualTrigger:
+        """Deserializes the ManualTrigger from a dictionary."""
+        return cls()
+
+
 class MaturityLevel(Enum):
     """Maturity level for EventDetails."""
 
@@ -1335,19 +1377,19 @@ class PipelineStateInfo:
 class PipelineTrigger:
     cron: Optional[CronTrigger] = None
 
-    manual: Optional[Any] = None
+    manual: Optional[ManualTrigger] = None
 
     def as_dict(self) -> dict:
         """Serializes the PipelineTrigger into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.cron: body['cron'] = self.cron.as_dict()
-        if self.manual: body['manual'] = self.manual
+        if self.manual: body['manual'] = self.manual.as_dict()
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> PipelineTrigger:
         """Deserializes the PipelineTrigger from a dictionary."""
-        return cls(cron=_from_dict(d, 'cron', CronTrigger), manual=d.get('manual', None))
+        return cls(cron=_from_dict(d, 'cron', CronTrigger), manual=_from_dict(d, 'manual', ManualTrigger))
 
 
 @dataclass
@@ -1503,6 +1545,20 @@ class StartUpdateResponse:
         return cls(update_id=d.get('update_id', None))
 
 
+@dataclass
+class StopPipelineResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the StopPipelineResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> StopPipelineResponse:
+        """Deserializes the StopPipelineResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdateInfo:
     cause: Optional[UpdateInfoCause] = None
@@ -2106,8 +2162,10 @@ class PipelinesAPI:
 
         headers = {'Accept': 'application/json', }
 
-        self._api.do('POST', f'/api/2.0/pipelines/{pipeline_id}/stop', headers=headers)
-        return Wait(self.wait_get_pipeline_idle, pipeline_id=pipeline_id)
+        op_response = self._api.do('POST', f'/api/2.0/pipelines/{pipeline_id}/stop', headers=headers)
+        return Wait(self.wait_get_pipeline_idle,
+                    response=StopPipelineResponse.from_dict(op_response),
+                    pipeline_id=pipeline_id)
 
     def stop_and_wait(self, pipeline_id: str, timeout=timedelta(minutes=20)) -> GetPipelineResponse:
         return self.stop(pipeline_id=pipeline_id).result(timeout=timeout)

databricks/sdk/service/provisioning.py

@@ -597,6 +597,20 @@ class CustomerManagedKey:
                    use_cases=_repeated_enum(d, 'use_cases', KeyUseCase))
 
 
+@dataclass
+class DeleteResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+        """Deserializes the DeleteResponse from a dictionary."""
+        return cls()
+
+
 class EndpointUseCase(Enum):
     """This enumeration represents the type of Databricks VPC [endpoint service] that was used when
     creating this VPC endpoint.
@@ -1062,6 +1076,20 @@ class PrivateAccessSettings:
                    region=d.get('region', None))
 
 
+@dataclass
+class ReplaceResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the ReplaceResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ReplaceResponse:
+        """Deserializes the ReplaceResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class RootBucketInfo:
     """Root S3 bucket information."""
@@ -1142,6 +1170,20 @@ class StsRole:
         return cls(external_id=d.get('external_id', None), role_arn=d.get('role_arn', None))
 
 
+@dataclass
+class UpdateResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateResponse:
+        """Deserializes the UpdateResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdateWorkspaceRequest:
     aws_region: Optional[str] = None
@@ -1162,8 +1204,6 @@ class UpdateWorkspaceRequest:
     is available only for updating failed workspaces."""
 
     network_connectivity_config_id: Optional[str] = None
-    """The ID of the network connectivity configuration object, which is the parent resource of this
-    private endpoint rule object."""
 
     network_id: Optional[str] = None
     """The ID of the workspace's network configuration object. Used only if you already use a
@@ -2637,7 +2677,10 @@ class WorkspacesAPI:
         the private access settings ID to upgrade a workspace to add support for front-end, back-end, or both
         types of connectivity. You cannot remove (downgrade) any existing front-end or back-end PrivateLink
         support on a workspace. - Custom tags. Given you provide an empty custom tags, the update would not be
-        applied.
+        applied. - Network connectivity configuration ID to add serverless stable IP support. You can add or
+        update the network connectivity configuration ID to ensure the workspace uses the same set of stable
+        IP CIDR blocks to access your resources. You cannot remove a network connectivity configuration from
+        the workspace once attached, you can only switch to another one.
         
         After calling the `PATCH` operation to update the workspace configuration, make repeated `GET`
         requests with the workspace ID and check the workspace status. The workspace is successful if the
@@ -2648,28 +2691,29 @@ class WorkspacesAPI:
         
         ### Update a running workspace You can update a Databricks workspace configuration for running
         workspaces for some fields, but not all fields. For a running workspace, this request supports
-        updating the following fields only: - Credential configuration ID
-        
-        - Network configuration ID. Used only if you already use a customer-managed VPC. You cannot convert a
-        running workspace from a Databricks-managed VPC to a customer-managed VPC. You can use a network
-        configuration update in this API for a failed or running workspace to add support for PrivateLink,
-        although you also need to add a private access settings object.
-        
-        - Key configuration ID for managed services (control plane storage, such as notebook source and
-        Databricks SQL queries). Databricks does not directly encrypt the data with the customer-managed key
-        (CMK). Databricks uses both the CMK and the Databricks managed key (DMK) that is unique to your
-        workspace to encrypt the Data Encryption Key (DEK). Databricks uses the DEK to encrypt your
-        workspace's managed services persisted data. If the workspace does not already have a CMK for managed
-        services, adding this ID enables managed services encryption for new or updated data. Existing managed
-        services data that existed before adding the key remains not encrypted with the DEK until it is
-        modified. If the workspace already has customer-managed keys for managed services, this request
-        rotates (changes) the CMK keys and the DEK is re-encrypted with the DMK and the new CMK. - Key
+        updating the following fields only: - Credential configuration ID - Network configuration ID. Used
+        only if you already use a customer-managed VPC. You cannot convert a running workspace from a
+        Databricks-managed VPC to a customer-managed VPC. You can use a network configuration update in this
+        API for a failed or running workspace to add support for PrivateLink, although you also need to add a
+        private access settings object. - Key configuration ID for managed services (control plane storage,
+        such as notebook source and Databricks SQL queries). Databricks does not directly encrypt the data
+        with the customer-managed key (CMK). Databricks uses both the CMK and the Databricks managed key (DMK)
+        that is unique to your workspace to encrypt the Data Encryption Key (DEK). Databricks uses the DEK to
+        encrypt your workspace's managed services persisted data. If the workspace does not already have a CMK
+        for managed services, adding this ID enables managed services encryption for new or updated data.
+        Existing managed services data that existed before adding the key remains not encrypted with the DEK
+        until it is modified. If the workspace already has customer-managed keys for managed services, this
+        request rotates (changes) the CMK keys and the DEK is re-encrypted with the DMK and the new CMK. - Key
         configuration ID for workspace storage (root S3 bucket and, optionally, EBS volumes). You can set this
         only if the workspace does not already have a customer-managed key configuration for workspace
         storage. - Private access settings ID to add PrivateLink support. You can add or update the private
         access settings ID to upgrade a workspace to add support for front-end, back-end, or both types of
         connectivity. You cannot remove (downgrade) any existing front-end or back-end PrivateLink support on
-        a workspace. - Custom tags. Given you provide an empty custom tags, the update would not be applied.
+        a workspace. - Custom tags. Given you provide an empty custom tags, the update would not be applied. -
+        Network connectivity configuration ID to add serverless stable IP support. You can add or update the
+        network connectivity configuration ID to ensure the workspace uses the same set of stable IP CIDR
+        blocks to access your resources. You cannot remove a network connectivity configuration from the
+        workspace once attached, you can only switch to another one.
         
         **Important**: To update a running workspace, your workspace must have no running compute resources
         that run in your workspace's VPC in the Classic data plane. For example, stop all all-purpose
@@ -2684,15 +2728,13 @@ class WorkspacesAPI:
         [Account Console]. However, you cannot use or create clusters for another 20 minutes after that status
         change. This results in a total of up to 40 minutes in which you cannot create clusters. If you create
         or use clusters before this time interval elapses, clusters do not launch successfully, fail, or could
-        cause other unexpected behavior.
-        
-        * For workspaces with a customer-managed VPC, the workspace status stays at status `RUNNING` and the
-        VPC change happens immediately. A change to the storage customer-managed key configuration ID might
-        take a few minutes to update, so continue to check the workspace until you observe that it has been
-        updated. If the update fails, the workspace might revert silently to its original configuration. After
-        the workspace has been updated, you cannot use or create clusters for another 20 minutes. If you
-        create or use clusters before this time interval elapses, clusters do not launch successfully, fail,
-        or could cause other unexpected behavior.
+        cause other unexpected behavior. * For workspaces with a customer-managed VPC, the workspace status
+        stays at status `RUNNING` and the VPC change happens immediately. A change to the storage
+        customer-managed key configuration ID might take a few minutes to update, so continue to check the
+        workspace until you observe that it has been updated. If the update fails, the workspace might revert
+        silently to its original configuration. After the workspace has been updated, you cannot use or create
+        clusters for another 20 minutes. If you create or use clusters before this time interval elapses,
+        clusters do not launch successfully, fail, or could cause other unexpected behavior.
         
         If you update the _storage_ customer-managed key configurations, it takes 20 minutes for the changes
         to fully take effect. During the 20 minute wait, it is important that you stop all REST API calls to
@@ -2725,8 +2767,6 @@ class WorkspacesAPI:
           The ID of the workspace's managed services encryption key configuration object. This parameter is
           available only for updating failed workspaces.
         :param network_connectivity_config_id: str (optional)
-          The ID of the network connectivity configuration object, which is the parent resource of this
-          private endpoint rule object.
         :param network_id: str (optional)
           The ID of the workspace's network configuration object. Used only if you already use a
           customer-managed VPC. For failed workspaces only, you can switch from a Databricks-managed VPC to a
@@ -2756,11 +2796,13 @@ class WorkspacesAPI:
             body['storage_customer_managed_key_id'] = storage_customer_managed_key_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('PATCH',
-                     f'/api/2.0/accounts/{self._api.account_id}/workspaces/{workspace_id}',
-                     body=body,
-                     headers=headers)
-        return Wait(self.wait_get_workspace_running, workspace_id=workspace_id)
+        op_response = self._api.do('PATCH',
+                                   f'/api/2.0/accounts/{self._api.account_id}/workspaces/{workspace_id}',
+                                   body=body,
+                                   headers=headers)
+        return Wait(self.wait_get_workspace_running,
+                    response=UpdateResponse.from_dict(op_response),
+                    workspace_id=workspace_id)
 
     def update_and_wait(
         self,

databricks/sdk/service/serving.py

@@ -449,6 +449,20 @@ class DeleteAppResponse:
         return cls(name=d.get('name', None))
 
 
+@dataclass
+class DeleteResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+        """Deserializes the DeleteResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeployAppRequest:
     manifest: AppManifest
@@ -751,6 +765,20 @@ class EndpointTag:
         return cls(key=d.get('key', None), value=d.get('value', None))
 
 
+@dataclass
+class ExportMetricsResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the ExportMetricsResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ExportMetricsResponse:
+        """Deserializes the ExportMetricsResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ExternalModel:
     provider: ExternalModelProvider