databricks-sdk 0.66.0__py3-none-any.whl → 0.68.0__py3-none-any.whl

databricks/sdk/service/qualitymonitorv2.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict
+from databricks.sdk.service._internal import _enum, _from_dict, _repeated_dict
 
 _LOG = logging.getLogger("databricks.sdk")
 

databricks/sdk/service/serving.py

@@ -13,8 +13,10 @@ from typing import Any, BinaryIO, Callable, Dict, Iterator, List, Optional
 
 import requests
 
+from databricks.sdk.service._internal import (Wait, _enum, _from_dict,
+                                              _repeated_dict)
+
 from ..errors import OperationFailed
-from ._internal import Wait, _enum, _from_dict, _repeated_dict
 
 _LOG = logging.getLogger("databricks.sdk")
 
@@ -3905,6 +3907,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 +4740,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/service/settings.py

@@ -7,7 +7,8 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict, _repeated_enum
+from databricks.sdk.service._internal import (_enum, _from_dict,
+                                              _repeated_dict, _repeated_enum)
 
 _LOG = logging.getLogger("databricks.sdk")
 
@@ -660,7 +661,8 @@ class ComplianceSecurityProfileSetting:
 
 
 class ComplianceStandard(Enum):
-    """Compliance stardard for SHIELD customers"""
+    """Compliance standard for SHIELD customers. See README.md for how instructions of how to add new
+    standards."""
 
     CANADA_PROTECTED_B = "CANADA_PROTECTED_B"
     CYBER_ESSENTIAL_PLUS = "CYBER_ESSENTIAL_PLUS"
@@ -668,6 +670,7 @@ class ComplianceStandard(Enum):
     FEDRAMP_IL5 = "FEDRAMP_IL5"
     FEDRAMP_MODERATE = "FEDRAMP_MODERATE"
     GERMANY_C5 = "GERMANY_C5"
+    GERMANY_TISAX = "GERMANY_TISAX"
     HIPAA = "HIPAA"
     HITRUST = "HITRUST"
     IRAP_PROTECTED = "IRAP_PROTECTED"

databricks/sdk/service/settingsv2.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict
+from databricks.sdk.service._internal import _enum, _from_dict, _repeated_dict
 
 _LOG = logging.getLogger("databricks.sdk")
 

databricks/sdk/service/sharing.py

@@ -7,13 +7,13 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict, _repeated_enum
+from databricks.sdk.service import catalog
+from databricks.sdk.service._internal import (_enum, _from_dict,
+                                              _repeated_dict, _repeated_enum)
 
 _LOG = logging.getLogger("databricks.sdk")
 
 
-from databricks.sdk.service import catalog
-
 # all definitions in this file are in alphabetical order
 
 
@@ -2310,6 +2310,10 @@ class TableInternalAttributes:
     auxiliary_managed_location: Optional[str] = None
     """Managed Delta Metadata location for foreign iceberg tables."""
 
+    dependency_storage_locations: Optional[List[str]] = None
+    """Storage locations of all table dependencies for shared views. Used on the recipient side for SEG
+    (Secure Egress Gateway) whitelisting."""
+
     parent_storage_location: Optional[str] = None
     """Will be populated in the reconciliation response for VIEW and FOREIGN_TABLE, with the value of
     the parent UC entity's storage_location, following the same logic as getManagedEntityPath in
@@ -2332,6 +2336,8 @@ class TableInternalAttributes:
         body = {}
         if self.auxiliary_managed_location is not None:
             body["auxiliary_managed_location"] = self.auxiliary_managed_location
+        if self.dependency_storage_locations:
+            body["dependency_storage_locations"] = [v for v in self.dependency_storage_locations]
         if self.parent_storage_location is not None:
             body["parent_storage_location"] = self.parent_storage_location
         if self.storage_location is not None:
@@ -2347,6 +2353,8 @@ class TableInternalAttributes:
         body = {}
         if self.auxiliary_managed_location is not None:
             body["auxiliary_managed_location"] = self.auxiliary_managed_location
+        if self.dependency_storage_locations:
+            body["dependency_storage_locations"] = self.dependency_storage_locations
         if self.parent_storage_location is not None:
             body["parent_storage_location"] = self.parent_storage_location
         if self.storage_location is not None:
@@ -2362,6 +2370,7 @@ class TableInternalAttributes:
         """Deserializes the TableInternalAttributes from a dictionary."""
         return cls(
             auxiliary_managed_location=d.get("auxiliary_managed_location", None),
+            dependency_storage_locations=d.get("dependency_storage_locations", None),
             parent_storage_location=d.get("parent_storage_location", None),
             storage_location=d.get("storage_location", None),
             type=_enum(d, "type", TableInternalAttributesSharedTableType),