databricks-sdk 0.70.0__py3-none-any.whl → 0.72.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

@@ -3363,6 +3363,7 @@ class EventDetailsCause(Enum):
 
     AUTORECOVERY = "AUTORECOVERY"
     AUTOSCALE = "AUTOSCALE"
+    AUTOSCALE_V2 = "AUTOSCALE_V2"
     REPLACE_BAD_NODES = "REPLACE_BAD_NODES"
     USER_REQUEST = "USER_REQUEST"
 
@@ -3377,6 +3378,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 +4826,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 +4858,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 +4871,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 +4884,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 +7105,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 +7802,7 @@ class ClusterPoliciesAPI:
 
         :returns: :class:`CreatePolicyResponse`
         """
+
         body = {}
         if definition is not None:
             body["definition"] = definition
@@ -7814,6 +7834,7 @@ class ClusterPoliciesAPI:
 
 
         """
+
         body = {}
         if policy_id is not None:
             body["policy_id"] = policy_id
@@ -7873,6 +7894,7 @@ class ClusterPoliciesAPI:
 
 
         """
+
         body = {}
         if definition is not None:
             body["definition"] = definition
@@ -7991,6 +8013,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 +8039,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 +8157,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -8332,6 +8357,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 +8523,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 +8722,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 +8919,7 @@ class ClustersAPI:
 
         :returns: Iterator over :class:`ClusterEvent`
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9064,6 +9093,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9082,6 +9112,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9117,6 +9148,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 +9186,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 +9217,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 +9257,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 +9281,7 @@ class ClustersAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -9290,6 +9326,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 +9363,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 +9507,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 +9600,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 +9632,7 @@ class CommandExecutionAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["clusterId"] = cluster_id
@@ -9628,6 +9669,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 +9743,7 @@ class GlobalInitScriptsAPI:
 
         :returns: :class:`CreateResponse`
         """
+
         body = {}
         if enabled is not None:
             body["enabled"] = enabled
@@ -9793,6 +9836,7 @@ class GlobalInitScriptsAPI:
 
 
         """
+
         body = {}
         if enabled is not None:
             body["enabled"] = enabled
@@ -9905,6 +9949,7 @@ class InstancePoolsAPI:
 
         :returns: :class:`CreateInstancePoolResponse`
         """
+
         body = {}
         if aws_attributes is not None:
             body["aws_attributes"] = aws_attributes.as_dict()
@@ -9952,6 +9997,7 @@ class InstancePoolsAPI:
 
 
         """
+
         body = {}
         if instance_pool_id is not None:
             body["instance_pool_id"] = instance_pool_id
@@ -10013,6 +10059,7 @@ class InstancePoolsAPI:
 
 
         """
+
         body = {}
         if custom_tags is not None:
             body["custom_tags"] = custom_tags
@@ -10120,6 +10167,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 +10191,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 +10254,7 @@ class InstanceProfilesAPI:
 
 
         """
+
         body = {}
         if iam_role_arn is not None:
             body["iam_role_arn"] = iam_role_arn
@@ -10259,6 +10309,7 @@ class InstanceProfilesAPI:
 
 
         """
+
         body = {}
         if iam_role_arn is not None:
             body["iam_role_arn"] = iam_role_arn
@@ -10301,6 +10352,7 @@ class InstanceProfilesAPI:
 
 
         """
+
         body = {}
         if instance_profile_arn is not None:
             body["instance_profile_arn"] = instance_profile_arn
@@ -10381,6 +10433,7 @@ class LibrariesAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -10404,6 +10457,7 @@ class LibrariesAPI:
 
 
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id
@@ -10453,6 +10507,7 @@ class PolicyComplianceForClustersAPI:
 
         :returns: :class:`EnforceClusterComplianceResponse`
         """
+
         body = {}
         if cluster_id is not None:
             body["cluster_id"] = cluster_id

databricks/sdk/service/dashboards.py

@@ -1189,6 +1189,7 @@ class MessageErrorType(Enum):
     TOO_MANY_TABLES_EXCEPTION = "TOO_MANY_TABLES_EXCEPTION"
     UNEXPECTED_REPLY_PROCESS_EXCEPTION = "UNEXPECTED_REPLY_PROCESS_EXCEPTION"
     UNKNOWN_AI_MODEL = "UNKNOWN_AI_MODEL"
+    UNSUPPORTED_CONVERSATION_TYPE_EXCEPTION = "UNSUPPORTED_CONVERSATION_TYPE_EXCEPTION"
     WAREHOUSE_ACCESS_MISSING_EXCEPTION = "WAREHOUSE_ACCESS_MISSING_EXCEPTION"
     WAREHOUSE_NOT_FOUND_EXCEPTION = "WAREHOUSE_NOT_FOUND_EXCEPTION"
 
@@ -1756,6 +1757,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 +2118,7 @@ class GenieAPI:
 
 
         """
+
         body = {}
         if rating is not None:
             body["rating"] = rating.value
@@ -2143,6 +2146,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 +2199,7 @@ class LakeviewAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = dashboard.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2214,6 +2219,7 @@ class LakeviewAPI:
 
         :returns: :class:`Schedule`
         """
+
         body = schedule.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2235,6 +2241,7 @@ class LakeviewAPI:
 
         :returns: :class:`Subscription`
         """
+
         body = subscription.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2530,6 +2537,7 @@ class LakeviewAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = {}
         if display_name is not None:
             body["display_name"] = display_name
@@ -2562,6 +2570,7 @@ class LakeviewAPI:
 
         :returns: :class:`PublishedDashboard`
         """
+
         body = {}
         if embed_credentials is not None:
             body["embed_credentials"] = embed_credentials
@@ -2614,6 +2623,7 @@ class LakeviewAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = dashboard.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2635,6 +2645,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: