databricks-sdk 0.66.0__tar.gz → 0.67.0__tar.gz

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/CHANGELOG.md

@@ -1,5 +1,20 @@
 # Version changelog
 
+## Release v0.67.0
+
+### API Changes
+* Add `update_notifications()` method for [w.serving_endpoints](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/serving/serving_endpoints.html) workspace-level service.
+* Add `parameters` field for `databricks.sdk.service.dashboards.GenieQueryAttachment`.
+* Add `database_instance_name` field for `databricks.sdk.service.database.CreateDatabaseInstanceRoleRequest`.
+* Add `custom_tags`, `effective_custom_tags`, `effective_usage_policy_id` and `usage_policy_id` fields for `databricks.sdk.service.database.DatabaseInstance`.
+* Add `effective_attributes` and `instance_name` fields for `databricks.sdk.service.database.DatabaseInstanceRole`.
+* Add `external_use_schema` enum value for `databricks.sdk.service.catalog.Privilege`.
+* Add `stream_native` enum value for `databricks.sdk.service.catalog.SystemType`.
+* Add `exceeded_max_token_length_exception` enum value for `databricks.sdk.service.dashboards.MessageErrorType`.
+* Change `name` field for `databricks.sdk.service.database.DatabaseInstanceRole` to be required.
+* [Breaking] Change `name` field for `databricks.sdk.service.database.DatabaseInstanceRole` to be required.
+
+
 ## Release v0.66.0
 
 ### New Features and Improvements

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-sdk
-Version: 0.66.0
+Version: 0.67.0
 Summary: Databricks SDK for Python (Beta)
 Project-URL: Documentation, https://databricks-sdk-py.readthedocs.io
 Keywords: databricks,sdk

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks/sdk/__init__.py

@@ -12,7 +12,6 @@ from databricks.sdk.mixins.compute import ClustersExt
 from databricks.sdk.mixins.files import DbfsExt, FilesExt
 from databricks.sdk.mixins.jobs import JobsExt
 from databricks.sdk.mixins.open_ai_client import ServingEndpointsExt
-from databricks.sdk.mixins.sharing import SharesExt
 from databricks.sdk.mixins.workspace import WorkspaceExt
 from databricks.sdk.service import agentbricks as pkg_agentbricks
 from databricks.sdk.service import apps as pkg_apps
@@ -358,7 +357,7 @@ class WorkspaceClient:
             self._api_client, serving_endpoints, serving_endpoints_data_plane_token_source
         )
         self._settings = pkg_settings.SettingsAPI(self._api_client)
-        self._shares = SharesExt(self._api_client)
+        self._shares = pkg_sharing.SharesAPI(self._api_client)
         self._statement_execution = pkg_sql.StatementExecutionAPI(self._api_client)
         self._storage_credentials = pkg_catalog.StorageCredentialsAPI(self._api_client)
         self._system_schemas = pkg_catalog.SystemSchemasAPI(self._api_client)
@@ -882,7 +881,7 @@ class WorkspaceClient:
         return self._settings
 
     @property
-    def shares(self) -> SharesExt:
+    def shares(self) -> pkg_sharing.SharesAPI:
         """A share is a container instantiated with :method:shares/create."""
         return self._shares
 

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks/sdk/service/catalog.py

@@ -7856,6 +7856,7 @@ class Privilege(Enum):
     CREATE_VOLUME = "CREATE_VOLUME"
     EXECUTE = "EXECUTE"
     EXECUTE_CLEAN_ROOM_TASK = "EXECUTE_CLEAN_ROOM_TASK"
+    EXTERNAL_USE_SCHEMA = "EXTERNAL_USE_SCHEMA"
     MANAGE = "MANAGE"
     MANAGE_ALLOWLIST = "MANAGE_ALLOWLIST"
     MODIFY = "MODIFY"
@@ -8539,7 +8540,7 @@ class Securable:
 
 
 class SecurableKind(Enum):
-    """Latest kind: CONNECTION_PALANTIR_OAUTH_M2M = 263; Next id:264"""
+    """Latest kind: CONNECTION_SHAREPOINT_OAUTH_M2M = 264; Next id:265"""
 
     TABLE_DB_STORAGE = "TABLE_DB_STORAGE"
     TABLE_DELTA = "TABLE_DELTA"
@@ -9003,6 +9004,7 @@ class SystemType(Enum):
     SAP = "SAP"
     SERVICENOW = "SERVICENOW"
     SNOWFLAKE = "SNOWFLAKE"
+    STREAM_NATIVE = "STREAM_NATIVE"
     TABLEAU = "TABLEAU"
     TERADATA = "TERADATA"
     WORKDAY = "WORKDAY"
@@ -10249,6 +10251,11 @@ class VolumeInfo:
     """The unique identifier of the volume"""
 
     volume_type: Optional[VolumeType] = None
+    """The type of the volume. An external volume is located in the specified external location. A
+    managed volume is located in the default location which is specified by the parent schema, or
+    the parent catalog, or the Metastore. [Learn more]
+    
+    [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external"""
 
     def as_dict(self) -> dict:
         """Serializes the VolumeInfo into a dictionary suitable for use as a JSON request body."""
@@ -10353,11 +10360,6 @@ class VolumeInfo:
 
 
 class VolumeType(Enum):
-    """The type of the volume. An external volume is located in the specified external location. A
-    managed volume is located in the default location which is specified by the parent schema, or
-    the parent catalog, or the Metastore. [Learn more]
-
-    [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external"""
 
     EXTERNAL = "EXTERNAL"
     MANAGED = "MANAGED"
@@ -15424,6 +15426,11 @@ class VolumesAPI:
         :param name: str
           The name of the volume
         :param volume_type: :class:`VolumeType`
+          The type of the volume. An external volume is located in the specified external location. A managed
+          volume is located in the default location which is specified by the parent schema, or the parent
+          catalog, or the Metastore. [Learn more]
+
+          [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external
         :param comment: str (optional)
           The comment attached to the volume
         :param storage_location: str (optional)
@@ -15482,7 +15489,7 @@ class VolumesAPI:
 
         The returned volumes are filtered based on the privileges of the calling user. For example, the
         metastore admin is able to list all the volumes. A regular user needs to be the owner or have the
-        **READ VOLUME** privilege on the volume to recieve the volumes in the response. For the latter case,
+        **READ VOLUME** privilege on the volume to receive the volumes in the response. For the latter case,
         the caller must also be the owner or have the **USE_CATALOG** privilege on the parent catalog and the
         **USE_SCHEMA** privilege on the parent schema.
 

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks/sdk/service/dashboards.py

@@ -714,6 +714,8 @@ class GenieQueryAttachment:
     last_updated_timestamp: Optional[int] = None
     """Time when the user updated the query last"""
 
+    parameters: Optional[List[QueryAttachmentParameter]] = None
+
     query: Optional[str] = None
     """AI generated SQL query"""
 
@@ -736,6 +738,8 @@ class GenieQueryAttachment:
             body["id"] = self.id
         if self.last_updated_timestamp is not None:
             body["last_updated_timestamp"] = self.last_updated_timestamp
+        if self.parameters:
+            body["parameters"] = [v.as_dict() for v in self.parameters]
         if self.query is not None:
             body["query"] = self.query
         if self.query_result_metadata:
@@ -755,6 +759,8 @@ class GenieQueryAttachment:
             body["id"] = self.id
         if self.last_updated_timestamp is not None:
             body["last_updated_timestamp"] = self.last_updated_timestamp
+        if self.parameters:
+            body["parameters"] = self.parameters
         if self.query is not None:
             body["query"] = self.query
         if self.query_result_metadata:
@@ -772,6 +778,7 @@ class GenieQueryAttachment:
             description=d.get("description", None),
             id=d.get("id", None),
             last_updated_timestamp=d.get("last_updated_timestamp", None),
+            parameters=_repeated_dict(d, "parameters", QueryAttachmentParameter),
             query=d.get("query", None),
             query_result_metadata=_from_dict(d, "query_result_metadata", GenieResultMetadata),
             statement_id=d.get("statement_id", None),
@@ -1135,6 +1142,7 @@ class MessageErrorType(Enum):
     DESCRIBE_QUERY_INVALID_SQL_ERROR = "DESCRIBE_QUERY_INVALID_SQL_ERROR"
     DESCRIBE_QUERY_TIMEOUT = "DESCRIBE_QUERY_TIMEOUT"
     DESCRIBE_QUERY_UNEXPECTED_FAILURE = "DESCRIBE_QUERY_UNEXPECTED_FAILURE"
+    EXCEEDED_MAX_TOKEN_LENGTH_EXCEPTION = "EXCEEDED_MAX_TOKEN_LENGTH_EXCEPTION"
     FUNCTIONS_NOT_AVAILABLE_EXCEPTION = "FUNCTIONS_NOT_AVAILABLE_EXCEPTION"
     FUNCTION_ARGUMENTS_INVALID_EXCEPTION = "FUNCTION_ARGUMENTS_INVALID_EXCEPTION"
     FUNCTION_ARGUMENTS_INVALID_JSON_EXCEPTION = "FUNCTION_ARGUMENTS_INVALID_JSON_EXCEPTION"
@@ -1260,6 +1268,42 @@ class PublishedDashboard:
         )
 
 
+@dataclass
+class QueryAttachmentParameter:
+    keyword: Optional[str] = None
+
+    sql_type: Optional[str] = None
+
+    value: Optional[str] = None
+
+    def as_dict(self) -> dict:
+        """Serializes the QueryAttachmentParameter into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.keyword is not None:
+            body["keyword"] = self.keyword
+        if self.sql_type is not None:
+            body["sql_type"] = self.sql_type
+        if self.value is not None:
+            body["value"] = self.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the QueryAttachmentParameter into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.keyword is not None:
+            body["keyword"] = self.keyword
+        if self.sql_type is not None:
+            body["sql_type"] = self.sql_type
+        if self.value is not None:
+            body["value"] = self.value
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> QueryAttachmentParameter:
+        """Deserializes the QueryAttachmentParameter from a dictionary."""
+        return cls(keyword=d.get("keyword", None), sql_type=d.get("sql_type", None), value=d.get("value", None))
+
+
 @dataclass
 class Result:
     is_truncated: Optional[bool] = None

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks/sdk/service/database.py

@@ -18,6 +18,38 @@ _LOG = logging.getLogger("databricks.sdk")
 # all definitions in this file are in alphabetical order
 
 
+@dataclass
+class CustomTag:
+    key: Optional[str] = None
+    """The key of the custom tag."""
+
+    value: Optional[str] = None
+    """The value of the custom tag."""
+
+    def as_dict(self) -> dict:
+        """Serializes the CustomTag into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.key is not None:
+            body["key"] = self.key
+        if self.value is not None:
+            body["value"] = self.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CustomTag into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.key is not None:
+            body["key"] = self.key
+        if self.value is not None:
+            body["value"] = self.value
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> CustomTag:
+        """Deserializes the CustomTag from a dictionary."""
+        return cls(key=d.get("key", None), value=d.get("value", None))
+
+
 @dataclass
 class DatabaseCatalog:
     name: str
@@ -125,9 +157,16 @@ class DatabaseInstance:
     creator: Optional[str] = None
     """The email of the creator of the instance."""
 
+    custom_tags: Optional[List[CustomTag]] = None
+    """Custom tags associated with the instance. This field is only included on create and update
+    responses."""
+
     effective_capacity: Optional[str] = None
     """Deprecated. The sku of the instance; this field will always match the value of capacity."""
 
+    effective_custom_tags: Optional[List[CustomTag]] = None
+    """The recorded custom tags associated with the instance."""
+
     effective_enable_pg_native_login: Optional[bool] = None
     """Whether the instance has PG native password login enabled."""
 
@@ -145,6 +184,9 @@ class DatabaseInstance:
     effective_stopped: Optional[bool] = None
     """Whether the instance is stopped."""
 
+    effective_usage_policy_id: Optional[str] = None
+    """The policy that is applied to the instance."""
+
     enable_pg_native_login: Optional[bool] = None
     """Whether to enable PG native password login on the instance. Defaults to false."""
 
@@ -184,6 +226,9 @@ class DatabaseInstance:
     uid: Optional[str] = None
     """An immutable UUID identifier for the instance."""
 
+    usage_policy_id: Optional[str] = None
+    """The desired usage policy to associate with the instance."""
+
     def as_dict(self) -> dict:
         """Serializes the DatabaseInstance into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -195,8 +240,12 @@ class DatabaseInstance:
             body["creation_time"] = self.creation_time
         if self.creator is not None:
             body["creator"] = self.creator
+        if self.custom_tags:
+            body["custom_tags"] = [v.as_dict() for v in self.custom_tags]
         if self.effective_capacity is not None:
             body["effective_capacity"] = self.effective_capacity
+        if self.effective_custom_tags:
+            body["effective_custom_tags"] = [v.as_dict() for v in self.effective_custom_tags]
         if self.effective_enable_pg_native_login is not None:
             body["effective_enable_pg_native_login"] = self.effective_enable_pg_native_login
         if self.effective_enable_readable_secondaries is not None:
@@ -207,6 +256,8 @@ class DatabaseInstance:
             body["effective_retention_window_in_days"] = self.effective_retention_window_in_days
         if self.effective_stopped is not None:
             body["effective_stopped"] = self.effective_stopped
+        if self.effective_usage_policy_id is not None:
+            body["effective_usage_policy_id"] = self.effective_usage_policy_id
         if self.enable_pg_native_login is not None:
             body["enable_pg_native_login"] = self.enable_pg_native_login
         if self.enable_readable_secondaries is not None:
@@ -231,6 +282,8 @@ class DatabaseInstance:
             body["stopped"] = self.stopped
         if self.uid is not None:
             body["uid"] = self.uid
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -244,8 +297,12 @@ class DatabaseInstance:
             body["creation_time"] = self.creation_time
         if self.creator is not None:
             body["creator"] = self.creator
+        if self.custom_tags:
+            body["custom_tags"] = self.custom_tags
         if self.effective_capacity is not None:
             body["effective_capacity"] = self.effective_capacity
+        if self.effective_custom_tags:
+            body["effective_custom_tags"] = self.effective_custom_tags
         if self.effective_enable_pg_native_login is not None:
             body["effective_enable_pg_native_login"] = self.effective_enable_pg_native_login
         if self.effective_enable_readable_secondaries is not None:
@@ -256,6 +313,8 @@ class DatabaseInstance:
             body["effective_retention_window_in_days"] = self.effective_retention_window_in_days
         if self.effective_stopped is not None:
             body["effective_stopped"] = self.effective_stopped
+        if self.effective_usage_policy_id is not None:
+            body["effective_usage_policy_id"] = self.effective_usage_policy_id
         if self.enable_pg_native_login is not None:
             body["enable_pg_native_login"] = self.enable_pg_native_login
         if self.enable_readable_secondaries is not None:
@@ -280,6 +339,8 @@ class DatabaseInstance:
             body["stopped"] = self.stopped
         if self.uid is not None:
             body["uid"] = self.uid
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         return body
 
     @classmethod
@@ -290,12 +351,15 @@ class DatabaseInstance:
             child_instance_refs=_repeated_dict(d, "child_instance_refs", DatabaseInstanceRef),
             creation_time=d.get("creation_time", None),
             creator=d.get("creator", None),
+            custom_tags=_repeated_dict(d, "custom_tags", CustomTag),
             effective_capacity=d.get("effective_capacity", None),
+            effective_custom_tags=_repeated_dict(d, "effective_custom_tags", CustomTag),
             effective_enable_pg_native_login=d.get("effective_enable_pg_native_login", None),
             effective_enable_readable_secondaries=d.get("effective_enable_readable_secondaries", None),
             effective_node_count=d.get("effective_node_count", None),
             effective_retention_window_in_days=d.get("effective_retention_window_in_days", None),
             effective_stopped=d.get("effective_stopped", None),
+            effective_usage_policy_id=d.get("effective_usage_policy_id", None),
             enable_pg_native_login=d.get("enable_pg_native_login", None),
             enable_readable_secondaries=d.get("enable_readable_secondaries", None),
             name=d.get("name", None),
@@ -308,6 +372,7 @@ class DatabaseInstance:
             state=_enum(d, "state", DatabaseInstanceState),
             stopped=d.get("stopped", None),
             uid=d.get("uid", None),
+            usage_policy_id=d.get("usage_policy_id", None),
         )
 
 
@@ -390,25 +455,34 @@ class DatabaseInstanceRef:
 class DatabaseInstanceRole:
     """A DatabaseInstanceRole represents a Postgres role in a database instance."""
 
+    name: str
+    """The name of the role. This is the unique identifier for the role in an instance."""
+
     attributes: Optional[DatabaseInstanceRoleAttributes] = None
-    """API-exposed Postgres role attributes"""
+    """The desired API-exposed Postgres role attribute to associate with the role. Optional."""
+
+    effective_attributes: Optional[DatabaseInstanceRoleAttributes] = None
+    """The attributes that are applied to the role."""
 
     identity_type: Optional[DatabaseInstanceRoleIdentityType] = None
     """The type of the role."""
 
+    instance_name: Optional[str] = None
+
     membership_role: Optional[DatabaseInstanceRoleMembershipRole] = None
     """An enum value for a standard role that this role is a member of."""
 
-    name: Optional[str] = None
-    """The name of the role. This is the unique identifier for the role in an instance."""
-
     def as_dict(self) -> dict:
         """Serializes the DatabaseInstanceRole into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.attributes:
             body["attributes"] = self.attributes.as_dict()
+        if self.effective_attributes:
+            body["effective_attributes"] = self.effective_attributes.as_dict()
         if self.identity_type is not None:
             body["identity_type"] = self.identity_type.value
+        if self.instance_name is not None:
+            body["instance_name"] = self.instance_name
         if self.membership_role is not None:
             body["membership_role"] = self.membership_role.value
         if self.name is not None:
@@ -420,8 +494,12 @@ class DatabaseInstanceRole:
         body = {}
         if self.attributes:
             body["attributes"] = self.attributes
+        if self.effective_attributes:
+            body["effective_attributes"] = self.effective_attributes
         if self.identity_type is not None:
             body["identity_type"] = self.identity_type
+        if self.instance_name is not None:
+            body["instance_name"] = self.instance_name
         if self.membership_role is not None:
             body["membership_role"] = self.membership_role
         if self.name is not None:
@@ -433,7 +511,9 @@ class DatabaseInstanceRole:
         """Deserializes the DatabaseInstanceRole from a dictionary."""
         return cls(
             attributes=_from_dict(d, "attributes", DatabaseInstanceRoleAttributes),
+            effective_attributes=_from_dict(d, "effective_attributes", DatabaseInstanceRoleAttributes),
             identity_type=_enum(d, "identity_type", DatabaseInstanceRoleIdentityType),
+            instance_name=d.get("instance_name", None),
             membership_role=_enum(d, "membership_role", DatabaseInstanceRoleMembershipRole),
             name=d.get("name", None),
         )
@@ -1527,22 +1607,32 @@ class DatabaseAPI:
         return self.create_database_instance(database_instance=database_instance).result(timeout=timeout)
 
     def create_database_instance_role(
-        self, instance_name: str, database_instance_role: DatabaseInstanceRole
+        self,
+        instance_name: str,
+        database_instance_role: DatabaseInstanceRole,
+        *,
+        database_instance_name: Optional[str] = None,
     ) -> DatabaseInstanceRole:
         """Create a role for a Database Instance.
 
         :param instance_name: str
         :param database_instance_role: :class:`DatabaseInstanceRole`
+        :param database_instance_name: str (optional)
 
         :returns: :class:`DatabaseInstanceRole`
         """
         body = database_instance_role.as_dict()
+        query = {}
+        if database_instance_name is not None:
+            query["database_instance_name"] = database_instance_name
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",
         }
 
-        res = self._api.do("POST", f"/api/2.0/database/instances/{instance_name}/roles", body=body, headers=headers)
+        res = self._api.do(
+            "POST", f"/api/2.0/database/instances/{instance_name}/roles", query=query, body=body, headers=headers
+        )
         return DatabaseInstanceRole.from_dict(res)
 
     def create_database_table(self, table: DatabaseTable) -> DatabaseTable:

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks/sdk/service/iam.py

@@ -3583,7 +3583,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str
@@ -3604,7 +3604,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str
@@ -3634,7 +3634,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str
@@ -3668,7 +3668,7 @@ class PermissionsAPI:
 
         :param request_object_type: str
           The type of the request object. Can be one of the following: alerts, alertsv2, authorization,
-          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files,
+          clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, genie,
           instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, or
           warehouses.
         :param request_object_id: str

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks/sdk/service/serving.py

@@ -3905,6 +3905,38 @@ class TrafficConfig:
         return cls(routes=_repeated_dict(d, "routes", Route))
 
 
+@dataclass
+class UpdateInferenceEndpointNotificationsResponse:
+    email_notifications: Optional[EmailNotifications] = None
+
+    name: Optional[str] = None
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateInferenceEndpointNotificationsResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.email_notifications:
+            body["email_notifications"] = self.email_notifications.as_dict()
+        if self.name is not None:
+            body["name"] = self.name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the UpdateInferenceEndpointNotificationsResponse into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.email_notifications:
+            body["email_notifications"] = self.email_notifications
+        if self.name is not None:
+            body["name"] = self.name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> UpdateInferenceEndpointNotificationsResponse:
+        """Deserializes the UpdateInferenceEndpointNotificationsResponse from a dictionary."""
+        return cls(
+            email_notifications=_from_dict(d, "email_notifications", EmailNotifications), name=d.get("name", None)
+        )
+
+
 @dataclass
 class V1ResponseChoiceElement:
     finish_reason: Optional[str] = None
@@ -4706,6 +4738,30 @@ class ServingEndpointsAPI:
             traffic_config=traffic_config,
         ).result(timeout=timeout)
 
+    def update_notifications(
+        self, name: str, *, email_notifications: Optional[EmailNotifications] = None
+    ) -> UpdateInferenceEndpointNotificationsResponse:
+        """Updates the email and webhook notification settings for an endpoint.
+
+        :param name: str
+          The name of the serving endpoint whose notifications are being updated. This field is required.
+        :param email_notifications: :class:`EmailNotifications` (optional)
+          The email notification settings to update. Specify email addresses to notify when endpoint state
+          changes occur.
+
+        :returns: :class:`UpdateInferenceEndpointNotificationsResponse`
+        """
+        body = {}
+        if email_notifications is not None:
+            body["email_notifications"] = email_notifications.as_dict()
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+        }
+
+        res = self._api.do("PATCH", f"/api/2.0/serving-endpoints/{name}/notifications", body=body, headers=headers)
+        return UpdateInferenceEndpointNotificationsResponse.from_dict(res)
+
     def update_permissions(
         self,
         serving_endpoint_id: str,

databricks_sdk-0.67.0/databricks/sdk/version.py

@@ -0,0 +1 @@
+__version__ = "0.67.0"

{databricks_sdk-0.66.0 → databricks_sdk-0.67.0}/databricks_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-sdk
-Version: 0.66.0
+Version: 0.67.0
 Summary: Databricks SDK for Python (Beta)
 Project-URL: Documentation, https://databricks-sdk-py.readthedocs.io
 Keywords: databricks,sdk

databricks_sdk-0.66.0/databricks/sdk/version.py

@@ -1 +0,0 @@
-__version__ = "0.66.0"