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

databricks/sdk/service/compute.py

@@ -64,6 +64,20 @@ class AddInstanceProfile:
                    skip_validation=d.get('skip_validation', None))
 
 
+@dataclass
+class AddResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the AddResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> AddResponse:
+        """Deserializes the AddResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Adlsgen2Info:
     destination: str
@@ -130,7 +144,8 @@ class AwsAttributes:
     will be overridden."""
 
     ebs_volume_iops: Optional[int] = None
-    """<needs content added>"""
+    """If using gp3 volumes, what IOPS to use for the disk. If this is not set, the maximum performance
+    of a gp2 volume with the same volume size will be used."""
 
     ebs_volume_size: Optional[int] = None
     """The size of each EBS volume (in GiB) launched for each instance. For general purpose SSD, this
@@ -138,7 +153,8 @@ class AwsAttributes:
     within the range 500 - 4096."""
 
     ebs_volume_throughput: Optional[int] = None
-    """<needs content added>"""
+    """If using gp3 volumes, what throughput to use for the disk. If this is not set, the maximum
+    performance of a gp2 volume with the same volume size will be used."""
 
     ebs_volume_type: Optional[EbsVolumeType] = None
     """The type of EBS volumes that will be launched with this cluster."""
@@ -302,6 +318,20 @@ class CancelCommand:
                    context_id=d.get('contextId', None))
 
 
+@dataclass
+class CancelResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the CancelResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CancelResponse:
+        """Deserializes the CancelResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ChangeClusterOwner:
     cluster_id: str
@@ -323,6 +353,20 @@ class ChangeClusterOwner:
         return cls(cluster_id=d.get('cluster_id', None), owner_username=d.get('owner_username', None))
 
 
+@dataclass
+class ChangeClusterOwnerResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the ChangeClusterOwnerResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ChangeClusterOwnerResponse:
+        """Deserializes the ChangeClusterOwnerResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ClientsTypes:
     jobs: Optional[bool] = None
@@ -2263,6 +2307,20 @@ class DeleteCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class DeleteClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteClusterResponse:
+        """Deserializes the DeleteClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeleteInstancePool:
     instance_pool_id: str
@@ -2280,6 +2338,20 @@ class DeleteInstancePool:
         return cls(instance_pool_id=d.get('instance_pool_id', None))
 
 
+@dataclass
+class DeleteInstancePoolResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteInstancePoolResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteInstancePoolResponse:
+        """Deserializes the DeleteInstancePoolResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DeletePolicy:
     policy_id: str
@@ -2297,6 +2369,34 @@ class DeletePolicy:
         return cls(policy_id=d.get('policy_id', None))
 
 
+@dataclass
+class DeletePolicyResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeletePolicyResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeletePolicyResponse:
+        """Deserializes the DeletePolicyResponse from a dictionary."""
+        return cls()
+
+
+@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 DestroyContext:
     cluster_id: str
@@ -2316,6 +2416,20 @@ class DestroyContext:
         return cls(cluster_id=d.get('clusterId', None), context_id=d.get('contextId', None))
 
 
+@dataclass
+class DestroyResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DestroyResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DestroyResponse:
+        """Deserializes the DestroyResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class DiskSpec:
     disk_count: Optional[int] = None
@@ -2670,6 +2784,20 @@ class EditCluster:
                    workload_type=_from_dict(d, 'workload_type', WorkloadType))
 
 
+@dataclass
+class EditClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditClusterResponse:
+        """Deserializes the EditClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EditInstancePool:
     instance_pool_id: str
@@ -2731,6 +2859,20 @@ class EditInstancePool:
                    node_type_id=d.get('node_type_id', None))
 
 
+@dataclass
+class EditInstancePoolResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditInstancePoolResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditInstancePoolResponse:
+        """Deserializes the EditInstancePoolResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EditPolicy:
     policy_id: str
@@ -2799,6 +2941,34 @@ class EditPolicy:
                    policy_id=d.get('policy_id', None))
 
 
+@dataclass
+class EditPolicyResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditPolicyResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditPolicyResponse:
+        """Deserializes the EditPolicyResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class EditResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditResponse:
+        """Deserializes the EditResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EventDetails:
     attributes: Optional[ClusterAttributes] = None
@@ -3686,6 +3856,20 @@ class InstallLibraries:
         return cls(cluster_id=d.get('cluster_id', None), libraries=_repeated_dict(d, 'libraries', Library))
 
 
+@dataclass
+class InstallLibrariesResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the InstallLibrariesResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> InstallLibrariesResponse:
+        """Deserializes the InstallLibrariesResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class InstancePoolAccessControlRequest:
     group_name: Optional[str] = None
@@ -4763,6 +4947,20 @@ class PermanentDeleteCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class PermanentDeleteClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the PermanentDeleteClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> PermanentDeleteClusterResponse:
+        """Deserializes the PermanentDeleteClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class PinCluster:
     cluster_id: str
@@ -4780,6 +4978,20 @@ class PinCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class PinClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the PinClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> PinClusterResponse:
+        """Deserializes the PinClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Policy:
     created_at_timestamp: Optional[int] = None
@@ -4955,6 +5167,20 @@ class RemoveInstanceProfile:
         return cls(instance_profile_arn=d.get('instance_profile_arn', None))
 
 
+@dataclass
+class RemoveResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RemoveResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RemoveResponse:
+        """Deserializes the RemoveResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ResizeCluster:
     cluster_id: str
@@ -4990,6 +5216,20 @@ class ResizeCluster:
                    num_workers=d.get('num_workers', None))
 
 
+@dataclass
+class ResizeClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the ResizeClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ResizeClusterResponse:
+        """Deserializes the ResizeClusterResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class RestartCluster:
     cluster_id: str
@@ -5011,6 +5251,20 @@ class RestartCluster:
         return cls(cluster_id=d.get('cluster_id', None), restart_user=d.get('restart_user', None))
 
 
+@dataclass
+class RestartClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RestartClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RestartClusterResponse:
+        """Deserializes the RestartClusterResponse from a dictionary."""
+        return cls()
+
+
 class ResultType(Enum):
 
     ERROR = 'error'
@@ -5261,6 +5515,20 @@ class StartCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class StartClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the StartClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> StartClusterResponse:
+        """Deserializes the StartClusterResponse from a dictionary."""
+        return cls()
+
+
 class State(Enum):
     """Current state of the cluster."""
 
@@ -5415,6 +5683,20 @@ class UninstallLibraries:
         return cls(cluster_id=d.get('cluster_id', None), libraries=_repeated_dict(d, 'libraries', Library))
 
 
+@dataclass
+class UninstallLibrariesResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UninstallLibrariesResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UninstallLibrariesResponse:
+        """Deserializes the UninstallLibrariesResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UnpinCluster:
     cluster_id: str
@@ -5432,6 +5714,34 @@ class UnpinCluster:
         return cls(cluster_id=d.get('cluster_id', None))
 
 
+@dataclass
+class UnpinClusterResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UnpinClusterResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UnpinClusterResponse:
+        """Deserializes the UnpinClusterResponse from a dictionary."""
+        return cls()
+
+
+@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 VolumesStorageInfo:
     destination: str
@@ -6164,8 +6474,10 @@ class ClustersAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('POST', '/api/2.0/clusters/delete', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_terminated, cluster_id=cluster_id)
+        op_response = self._api.do('POST', '/api/2.0/clusters/delete', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_terminated,
+                    response=DeleteClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def delete_and_wait(self, cluster_id: str, timeout=timedelta(minutes=20)) -> ClusterDetails:
         return self.delete(cluster_id=cluster_id).result(timeout=timeout)
@@ -6369,8 +6681,10 @@ class ClustersAPI:
         if workload_type is not None: body['workload_type'] = workload_type.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('POST', '/api/2.0/clusters/edit', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+        op_response = self._api.do('POST', '/api/2.0/clusters/edit', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=EditClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def edit_and_wait(
         self,
@@ -6668,8 +6982,10 @@ class ClustersAPI:
         if num_workers is not None: body['num_workers'] = num_workers
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('POST', '/api/2.0/clusters/resize', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+        op_response = self._api.do('POST', '/api/2.0/clusters/resize', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=ResizeClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def resize_and_wait(self,
                         cluster_id: str,
@@ -6700,8 +7016,10 @@ class ClustersAPI:
         if restart_user is not None: body['restart_user'] = restart_user
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('POST', '/api/2.0/clusters/restart', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+        op_response = self._api.do('POST', '/api/2.0/clusters/restart', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=RestartClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def restart_and_wait(self,
                          cluster_id: str,
@@ -6767,8 +7085,10 @@ class ClustersAPI:
         if cluster_id is not None: body['cluster_id'] = cluster_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('POST', '/api/2.0/clusters/start', body=body, headers=headers)
-        return Wait(self.wait_get_cluster_running, cluster_id=cluster_id)
+        op_response = self._api.do('POST', '/api/2.0/clusters/start', body=body, headers=headers)
+        return Wait(self.wait_get_cluster_running,
+                    response=StartClusterResponse.from_dict(op_response),
+                    cluster_id=cluster_id)
 
     def start_and_wait(self, cluster_id: str, timeout=timedelta(minutes=20)) -> ClusterDetails:
         return self.start(cluster_id=cluster_id).result(timeout=timeout)
@@ -6946,8 +7266,9 @@ class CommandExecutionAPI:
         if context_id is not None: body['contextId'] = context_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
 
-        self._api.do('POST', '/api/1.2/commands/cancel', body=body, headers=headers)
+        op_response = self._api.do('POST', '/api/1.2/commands/cancel', body=body, headers=headers)
         return Wait(self.wait_command_status_command_execution_cancelled,
+                    response=CancelResponse.from_dict(op_response),
                     cluster_id=cluster_id,
                     command_id=command_id,
                     context_id=context_id)

databricks/sdk/service/dashboards.py

@@ -39,6 +39,20 @@ class PublishRequest:
                    warehouse_id=d.get('warehouse_id', None))
 
 
+@dataclass
+class PublishResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the PublishResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> PublishResponse:
+        """Deserializes the PublishResponse from a dictionary."""
+        return cls()
+
+
 class LakeviewAPI:
     """These APIs provide specific management operations for Lakeview dashboards. Generic resource management can
     be done with Workspace API (import, export, get-status, list, delete)."""