databricks-sdk 0.70.0__py3-none-any.whl → 0.71.0__py3-none-any.whl

databricks/sdk/service/cleanrooms.py

@@ -12,7 +12,7 @@ from typing import Any, Callable, Dict, Iterator, List, Optional
 
 from databricks.sdk.service import catalog, jobs, settings, sharing
 from databricks.sdk.service._internal import (Wait, _enum, _from_dict,
-                                              _repeated_dict)
+                                              _repeated_dict, _repeated_enum)
 
 _LOG = logging.getLogger("databricks.sdk")
 
@@ -1080,7 +1080,7 @@ class ComplianceSecurityProfile:
         """Serializes the ComplianceSecurityProfile into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.compliance_standards:
-            body["compliance_standards"] = [v.as_dict() for v in self.compliance_standards]
+            body["compliance_standards"] = [v.value for v in self.compliance_standards]
         if self.is_enabled is not None:
             body["is_enabled"] = self.is_enabled
         return body
@@ -1098,7 +1098,7 @@ class ComplianceSecurityProfile:
     def from_dict(cls, d: Dict[str, Any]) -> ComplianceSecurityProfile:
         """Deserializes the ComplianceSecurityProfile from a dictionary."""
         return cls(
-            compliance_standards=_repeated_dict(d, "compliance_standards", settings.ComplianceStandard),
+            compliance_standards=_repeated_enum(d, "compliance_standards", settings.ComplianceStandard),
             is_enabled=d.get("is_enabled", None),
         )
 
@@ -1495,6 +1495,7 @@ class CleanRoomAssetsAPI:
 
         :returns: :class:`CleanRoomAsset`
         """
+
         body = asset.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1524,6 +1525,7 @@ class CleanRoomAssetsAPI:
 
         :returns: :class:`CreateCleanRoomAssetReviewResponse`
         """
+
         body = {}
         if notebook_review is not None:
             body["notebook_review"] = notebook_review.as_dict()
@@ -1635,6 +1637,7 @@ class CleanRoomAssetsAPI:
 
         :returns: :class:`CleanRoomAsset`
         """
+
         body = asset.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1666,6 +1669,7 @@ class CleanRoomAutoApprovalRulesAPI:
 
         :returns: :class:`CleanRoomAutoApprovalRule`
         """
+
         body = {}
         if auto_approval_rule is not None:
             body["auto_approval_rule"] = auto_approval_rule.as_dict()
@@ -1760,6 +1764,7 @@ class CleanRoomAutoApprovalRulesAPI:
 
         :returns: :class:`CleanRoomAutoApprovalRule`
         """
+
         body = auto_approval_rule.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1869,6 +1874,7 @@ class CleanRoomsAPI:
           Long-running operation waiter for :class:`CleanRoom`.
           See :method:wait_get_clean_room_active for more details.
         """
+
         body = clean_room.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1894,6 +1900,7 @@ class CleanRoomsAPI:
 
         :returns: :class:`CreateCleanRoomOutputCatalogResponse`
         """
+
         body = output_catalog.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1979,6 +1986,7 @@ class CleanRoomsAPI:
 
         :returns: :class:`CleanRoom`
         """
+
         body = {}
         if clean_room is not None:
             body["clean_room"] = clean_room.as_dict()

databricks/sdk/service/compute.py

@@ -3377,6 +3377,8 @@ class EventType(Enum):
     CLUSTER_MIGRATED = "CLUSTER_MIGRATED"
     CREATING = "CREATING"
     DBFS_DOWN = "DBFS_DOWN"
+    DECOMMISSION_ENDED = "DECOMMISSION_ENDED"
+    DECOMMISSION_STARTED = "DECOMMISSION_STARTED"
     DID_NOT_EXPAND_DISK = "DID_NOT_EXPAND_DISK"
     DRIVER_HEALTHY = "DRIVER_HEALTHY"
     DRIVER_NOT_RESPONDING = "DRIVER_NOT_RESPONDING"
@@ -4823,6 +4825,16 @@ class InstancePoolAwsAttributes:
     availability: Optional[InstancePoolAwsAttributesAvailability] = None
     """Availability type used for the spot nodes."""
 
+    instance_profile_arn: Optional[str] = None
+    """All AWS instances belonging to the instance pool will have this instance profile. If omitted,
+    instances will initially be launched with the workspace's default instance profile. If defined,
+    clusters that use the pool will inherit the instance profile, and must not specify their own
+    instance profile on cluster creation or update. If the pool does not specify an instance
+    profile, clusters using the pool may specify any instance profile. The instance profile must
+    have previously been added to the Databricks environment by an account administrator.
+    
+    This feature may only be available to certain customer plans."""
+
     spot_bid_price_percent: Optional[int] = None
     """Calculates the bid price for AWS spot instances, as a percentage of the corresponding instance
     type's on-demand price. For example, if this field is set to 50, and the cluster needs a new
@@ -4845,6 +4857,8 @@ class InstancePoolAwsAttributes:
         body = {}
         if self.availability is not None:
             body["availability"] = self.availability.value
+        if self.instance_profile_arn is not None:
+            body["instance_profile_arn"] = self.instance_profile_arn
         if self.spot_bid_price_percent is not None:
             body["spot_bid_price_percent"] = self.spot_bid_price_percent
         if self.zone_id is not None:
@@ -4856,6 +4870,8 @@ class InstancePoolAwsAttributes:
         body = {}
         if self.availability is not None:
             body["availability"] = self.availability
+        if self.instance_profile_arn is not None:
+            body["instance_profile_arn"] = self.instance_profile_arn
         if self.spot_bid_price_percent is not None:
             body["spot_bid_price_percent"] = self.spot_bid_price_percent
         if self.zone_id is not None:
@@ -4867,6 +4883,7 @@ class InstancePoolAwsAttributes:
         """Deserializes the InstancePoolAwsAttributes from a dictionary."""
         return cls(
             availability=_enum(d, "availability", InstancePoolAwsAttributesAvailability),
+            instance_profile_arn=d.get("instance_profile_arn", None),
             spot_bid_price_percent=d.get("spot_bid_price_percent", None),
             zone_id=d.get("zone_id", None),
         )
@@ -7087,6 +7104,7 @@ class TerminationReasonCode(Enum):
     DATABASE_CONNECTION_FAILURE = "DATABASE_CONNECTION_FAILURE"
     DATA_ACCESS_CONFIG_CHANGED = "DATA_ACCESS_CONFIG_CHANGED"
     DBFS_COMPONENT_UNHEALTHY = "DBFS_COMPONENT_UNHEALTHY"
+    DBR_IMAGE_RESOLUTION_FAILURE = "DBR_IMAGE_RESOLUTION_FAILURE"
     DISASTER_RECOVERY_REPLICATION = "DISASTER_RECOVERY_REPLICATION"
     DNS_RESOLUTION_ERROR = "DNS_RESOLUTION_ERROR"
     DOCKER_CONTAINER_CREATION_EXCEPTION = "DOCKER_CONTAINER_CREATION_EXCEPTION"
@@ -7783,6 +7801,7 @@ class ClusterPoliciesAPI:
 
         :returns: :class:`CreatePolicyResponse`
         """
+
         body = {}
         if definition is not None:
             body["definition"] = definition
@@ -7814,6 +7833,7 @@ class ClusterPoliciesAPI:
 
 
         """
+
         body = {}
         if policy_id is not None:
             body["policy_id"] = policy_id
@@ -7873,6 +7893,7 @@ class ClusterPoliciesAPI:
 
 
         """
+
         body = {}
         if definition is not None:
             body["definition"] = definition
@@ -7991,6 +8012,7 @@ class ClusterPoliciesAPI:
 
         :returns: :class:`ClusterPolicyPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -8016,6 +8038,7 @@ class ClusterPoliciesAPI:
 
         :returns: :class:`ClusterPolicyPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -8133,6 +8156,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -8332,6 +8356,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_running for more details.
         """
+
         body = {}
         if apply_policy_default_values is not None:
             body["apply_policy_default_values"] = apply_policy_default_values
@@ -8497,6 +8522,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_terminated for more details.
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -8695,6 +8721,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_running for more details.
         """
+
         body = {}
         if apply_policy_default_values is not None:
             body["apply_policy_default_values"] = apply_policy_default_values
@@ -8891,6 +8918,7 @@ class ClustersAPI:
 
         :returns: Iterator over :class:`ClusterEvent`
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9064,6 +9092,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9082,6 +9111,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9117,6 +9147,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_running for more details.
         """
+
         body = {}
         if autoscale is not None:
             body["autoscale"] = autoscale.as_dict()
@@ -9154,6 +9185,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_running for more details.
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9184,6 +9216,7 @@ class ClustersAPI:
 
         :returns: :class:`ClusterPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -9223,6 +9256,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_running for more details.
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9246,6 +9280,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9290,6 +9325,7 @@ class ClustersAPI:
           Long-running operation waiter for :class:`ClusterDetails`.
           See :method:wait_get_cluster_running for more details.
         """
+
         body = {}
         if cluster is not None:
             body["cluster"] = cluster.as_dict()
@@ -9326,6 +9362,7 @@ class ClustersAPI:
 
         :returns: :class:`ClusterPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -9469,6 +9506,7 @@ class CommandExecutionAPI:
           Long-running operation waiter for :class:`CommandStatusResponse`.
           See :method:wait_command_status_command_execution_cancelled for more details.
         """
+
         body = {}
         if cluster_id is not None:
             body["clusterId"] = cluster_id
@@ -9561,6 +9599,7 @@ class CommandExecutionAPI:
           Long-running operation waiter for :class:`ContextStatusResponse`.
           See :method:wait_context_status_command_execution_running for more details.
         """
+
         body = {}
         if cluster_id is not None:
             body["clusterId"] = cluster_id
@@ -9592,6 +9631,7 @@ class CommandExecutionAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["clusterId"] = cluster_id
@@ -9628,6 +9668,7 @@ class CommandExecutionAPI:
           Long-running operation waiter for :class:`CommandStatusResponse`.
           See :method:wait_command_status_command_execution_finished_or_error for more details.
         """
+
         body = {}
         if cluster_id is not None:
             body["clusterId"] = cluster_id
@@ -9701,6 +9742,7 @@ class GlobalInitScriptsAPI:
 
         :returns: :class:`CreateResponse`
         """
+
         body = {}
         if enabled is not None:
             body["enabled"] = enabled
@@ -9793,6 +9835,7 @@ class GlobalInitScriptsAPI:
 
 
         """
+
         body = {}
         if enabled is not None:
             body["enabled"] = enabled
@@ -9905,6 +9948,7 @@ class InstancePoolsAPI:
 
         :returns: :class:`CreateInstancePoolResponse`
         """
+
         body = {}
         if aws_attributes is not None:
             body["aws_attributes"] = aws_attributes.as_dict()
@@ -9952,6 +9996,7 @@ class InstancePoolsAPI:
 
 
         """
+
         body = {}
         if instance_pool_id is not None:
             body["instance_pool_id"] = instance_pool_id
@@ -10013,6 +10058,7 @@ class InstancePoolsAPI:
 
 
         """
+
         body = {}
         if custom_tags is not None:
             body["custom_tags"] = custom_tags
@@ -10120,6 +10166,7 @@ class InstancePoolsAPI:
 
         :returns: :class:`InstancePoolPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -10143,6 +10190,7 @@ class InstancePoolsAPI:
 
         :returns: :class:`InstancePoolPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -10205,6 +10253,7 @@ class InstanceProfilesAPI:
 
 
         """
+
         body = {}
         if iam_role_arn is not None:
             body["iam_role_arn"] = iam_role_arn
@@ -10259,6 +10308,7 @@ class InstanceProfilesAPI:
 
 
         """
+
         body = {}
         if iam_role_arn is not None:
             body["iam_role_arn"] = iam_role_arn
@@ -10301,6 +10351,7 @@ class InstanceProfilesAPI:
 
 
         """
+
         body = {}
         if instance_profile_arn is not None:
             body["instance_profile_arn"] = instance_profile_arn
@@ -10381,6 +10432,7 @@ class LibrariesAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -10404,6 +10456,7 @@ class LibrariesAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -10453,6 +10506,7 @@ class PolicyComplianceForClustersAPI:
 
         :returns: :class:`EnforceClusterComplianceResponse`
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id

databricks/sdk/service/dashboards.py

@@ -1756,6 +1756,7 @@ class GenieAPI:
           Long-running operation waiter for :class:`GenieMessage`.
           See :method:wait_get_message_genie_completed for more details.
         """
+
         body = {}
         if content is not None:
             body["content"] = content
@@ -2116,6 +2117,7 @@ class GenieAPI:
 
 
         """
+
         body = {}
         if rating is not None:
             body["rating"] = rating.value
@@ -2143,6 +2145,7 @@ class GenieAPI:
           Long-running operation waiter for :class:`GenieMessage`.
           See :method:wait_get_message_genie_completed for more details.
         """
+
         body = {}
         if content is not None:
             body["content"] = content
@@ -2195,6 +2198,7 @@ class LakeviewAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = dashboard.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2214,6 +2218,7 @@ class LakeviewAPI:
 
         :returns: :class:`Schedule`
         """
+
         body = schedule.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2235,6 +2240,7 @@ class LakeviewAPI:
 
         :returns: :class:`Subscription`
         """
+
         body = subscription.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2530,6 +2536,7 @@ class LakeviewAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = {}
         if display_name is not None:
             body["display_name"] = display_name
@@ -2562,6 +2569,7 @@ class LakeviewAPI:
 
         :returns: :class:`PublishedDashboard`
         """
+
         body = {}
         if embed_credentials is not None:
             body["embed_credentials"] = embed_credentials
@@ -2614,6 +2622,7 @@ class LakeviewAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = dashboard.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2635,6 +2644,7 @@ class LakeviewAPI:
 
         :returns: :class:`Schedule`
         """
+
         body = schedule.as_dict()
         headers = {
             "Accept": "application/json",

databricks/sdk/service/database.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 import logging
 import random
 import time
+import uuid
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
@@ -1570,6 +1571,7 @@ class DatabaseAPI:
 
         :returns: :class:`DatabaseCatalog`
         """
+
         body = catalog.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1589,6 +1591,7 @@ class DatabaseAPI:
           Long-running operation waiter for :class:`DatabaseInstance`.
           See :method:wait_get_database_instance_database_available for more details.
         """
+
         body = database_instance.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1622,6 +1625,7 @@ class DatabaseAPI:
 
         :returns: :class:`DatabaseInstanceRole`
         """
+
         body = database_instance_role.as_dict()
         query = {}
         if database_instance_name is not None:
@@ -1644,6 +1648,7 @@ class DatabaseAPI:
 
         :returns: :class:`DatabaseTable`
         """
+
         body = table.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1660,6 +1665,7 @@ class DatabaseAPI:
 
         :returns: :class:`SyncedDatabaseTable`
         """
+
         body = synced_table.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1806,6 +1812,9 @@ class DatabaseAPI:
 
         :returns: :class:`DatabaseCredential`
         """
+
+        if request_id is None or request_id == "":
+            request_id = str(uuid.uuid4())
         body = {}
         if claims is not None:
             body["claims"] = [v.as_dict() for v in claims]
@@ -2049,6 +2058,7 @@ class DatabaseAPI:
 
         :returns: :class:`DatabaseCatalog`
         """
+
         body = database_catalog.as_dict()
         query = {}
         if update_mask is not None:
@@ -2075,6 +2085,7 @@ class DatabaseAPI:
 
         :returns: :class:`DatabaseInstance`
         """
+
         body = database_instance.as_dict()
         query = {}
         if update_mask is not None:
@@ -2101,6 +2112,7 @@ class DatabaseAPI:
 
         :returns: :class:`SyncedDatabaseTable`
         """
+
         body = synced_table.as_dict()
         query = {}
         if update_mask is not None:

databricks/sdk/service/dataquality.py

@@ -909,6 +909,7 @@ class DataQualityAPI:
 
         :returns: :class:`Monitor`
         """
+
         body = monitor.as_dict()
         headers = {
             "Accept": "application/json",
@@ -946,6 +947,7 @@ class DataQualityAPI:
 
         :returns: :class:`Refresh`
         """
+
         body = refresh.as_dict()
         headers = {
             "Accept": "application/json",
@@ -1229,6 +1231,7 @@ class DataQualityAPI:
 
         :returns: :class:`Monitor`
         """
+
         body = monitor.as_dict()
         query = {}
         if update_mask is not None:
@@ -1271,6 +1274,7 @@ class DataQualityAPI:
 
         :returns: :class:`Refresh`
         """
+
         body = refresh.as_dict()
         query = {}
         if update_mask is not None: