databricks-sdk 0.53.0__py3-none-any.whl → 0.54.0__py3-none-any.whl

databricks/sdk/__init__.py

@@ -23,6 +23,7 @@ from databricks.sdk.service.catalog import (AccountMetastoreAssignmentsAPI,
                                             AccountStorageCredentialsAPI,
                                             ArtifactAllowlistsAPI, CatalogsAPI,
                                             ConnectionsAPI, CredentialsAPI,
+                                            DatabaseInstancesAPI,
                                             ExternalLocationsAPI, FunctionsAPI,
                                             GrantsAPI, MetastoresAPI,
                                             ModelVersionsAPI, OnlineTablesAPI,
@@ -89,12 +90,15 @@ from databricks.sdk.service.settings import (
     DisableLegacyDbfsAPI, DisableLegacyFeaturesAPI, EnableExportNotebookAPI,
     EnableIpAccessListsAPI, EnableNotebookTableClipboardAPI,
     EnableResultsDownloadingAPI, EnhancedSecurityMonitoringAPI,
-    EsmEnablementAccountAPI, IpAccessListsAPI, NetworkConnectivityAPI,
-    NotificationDestinationsAPI, PersonalComputeAPI,
+    EsmEnablementAccountAPI, IpAccessListsAPI,
+    LlmProxyPartnerPoweredAccountAPI, LlmProxyPartnerPoweredEnforceAPI,
+    LlmProxyPartnerPoweredWorkspaceAPI, NetworkConnectivityAPI,
+    NetworkPoliciesAPI, NotificationDestinationsAPI, PersonalComputeAPI,
     RestrictWorkspaceAdminsAPI, SettingsAPI, TokenManagementAPI, TokensAPI,
-    WorkspaceConfAPI)
+    WorkspaceConfAPI, WorkspaceNetworkConfigurationAPI)
 from databricks.sdk.service.sharing import (ProvidersAPI,
                                             RecipientActivationAPI,
+                                            RecipientFederationPoliciesAPI,
                                             RecipientsAPI, SharesAPI)
 from databricks.sdk.service.sql import (AlertsAPI, AlertsLegacyAPI,
                                         AlertsV2API, DashboardsAPI,
@@ -233,6 +237,7 @@ class WorkspaceClient:
         self._dashboard_widgets = service.sql.DashboardWidgetsAPI(self._api_client)
         self._dashboards = service.sql.DashboardsAPI(self._api_client)
         self._data_sources = service.sql.DataSourcesAPI(self._api_client)
+        self._database_instances = service.catalog.DatabaseInstancesAPI(self._api_client)
         self._dbfs = DbfsExt(self._api_client)
         self._dbsql_permissions = service.sql.DbsqlPermissionsAPI(self._api_client)
         self._experiments = service.ml.ExperimentsAPI(self._api_client)
@@ -282,6 +287,7 @@ class WorkspaceClient:
         self._query_visualizations = service.sql.QueryVisualizationsAPI(self._api_client)
         self._query_visualizations_legacy = service.sql.QueryVisualizationsLegacyAPI(self._api_client)
         self._recipient_activation = service.sharing.RecipientActivationAPI(self._api_client)
+        self._recipient_federation_policies = service.sharing.RecipientFederationPoliciesAPI(self._api_client)
         self._recipients = service.sharing.RecipientsAPI(self._api_client)
         self._redash_config = service.sql.RedashConfigAPI(self._api_client)
         self._registered_models = service.catalog.RegisteredModelsAPI(self._api_client)
@@ -459,6 +465,11 @@ class WorkspaceClient:
         """This API is provided to assist you in making new query objects."""
         return self._data_sources
 
+    @property
+    def database_instances(self) -> service.catalog.DatabaseInstancesAPI:
+        """Database Instances provide access to a database via REST API or direct SQL."""
+        return self._database_instances
+
     @property
     def dbfs(self) -> DbfsExt:
         """DBFS API makes it simple to interact with various data sources without having to include a users credentials every time to read a file."""
@@ -684,6 +695,11 @@ class WorkspaceClient:
         """The Recipient Activation API is only applicable in the open sharing model where the recipient object has the authentication type of `TOKEN`."""
         return self._recipient_activation
 
+    @property
+    def recipient_federation_policies(self) -> service.sharing.RecipientFederationPoliciesAPI:
+        """The Recipient Federation Policies APIs are only applicable in the open sharing model where the recipient object has the authentication type of `OIDC_RECIPIENT`, enabling data sharing from Databricks to non-Databricks recipients."""
+        return self._recipient_federation_policies
+
     @property
     def recipients(self) -> service.sharing.RecipientsAPI:
         """A recipient is an object you create using :method:recipients/create to represent an organization which you want to allow access shares."""
@@ -916,6 +932,7 @@ class AccountClient:
         self._metastore_assignments = service.catalog.AccountMetastoreAssignmentsAPI(self._api_client)
         self._metastores = service.catalog.AccountMetastoresAPI(self._api_client)
         self._network_connectivity = service.settings.NetworkConnectivityAPI(self._api_client)
+        self._network_policies = service.settings.NetworkPoliciesAPI(self._api_client)
         self._networks = service.provisioning.NetworksAPI(self._api_client)
         self._o_auth_published_apps = service.oauth2.OAuthPublishedAppsAPI(self._api_client)
         self._private_access = service.provisioning.PrivateAccessAPI(self._api_client)
@@ -930,6 +947,7 @@ class AccountClient:
         self._users = service.iam.AccountUsersAPI(self._api_client)
         self._vpc_endpoints = service.provisioning.VpcEndpointsAPI(self._api_client)
         self._workspace_assignment = service.iam.WorkspaceAssignmentAPI(self._api_client)
+        self._workspace_network_configuration = service.settings.WorkspaceNetworkConfigurationAPI(self._api_client)
         self._workspaces = service.provisioning.WorkspacesAPI(self._api_client)
         self._budgets = service.billing.BudgetsAPI(self._api_client)
 
@@ -1006,6 +1024,11 @@ class AccountClient:
         """These APIs provide configurations for the network connectivity of your workspaces for serverless compute resources."""
         return self._network_connectivity
 
+    @property
+    def network_policies(self) -> service.settings.NetworkPoliciesAPI:
+        """These APIs manage network policies for this account."""
+        return self._network_policies
+
     @property
     def networks(self) -> service.provisioning.NetworksAPI:
         """These APIs manage network configurations for customer-managed VPCs (optional)."""
@@ -1076,6 +1099,11 @@ class AccountClient:
         """The Workspace Permission Assignment API allows you to manage workspace permissions for principals in your account."""
         return self._workspace_assignment
 
+    @property
+    def workspace_network_configuration(self) -> service.settings.WorkspaceNetworkConfigurationAPI:
+        """These APIs allow configuration of network settings for Databricks workspaces."""
+        return self._workspace_network_configuration
+
     @property
     def workspaces(self) -> service.provisioning.WorkspacesAPI:
         """These APIs manage workspaces for this account."""

databricks/sdk/errors/base.py

@@ -1,5 +1,4 @@
 import re
-import warnings
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional
 
@@ -62,31 +61,6 @@ class DatabricksError(IOError):
         :param details:
         :param kwargs:
         """
-        # SCIM-specific parameters are deprecated.
-        if detail:
-            warnings.warn(
-                "The 'detail' parameter of DatabricksError is deprecated and will be removed in a future version."
-            )
-        if scimType:
-            warnings.warn(
-                "The 'scimType' parameter of DatabricksError is deprecated and will be removed in a future version."
-            )
-        if status:
-            warnings.warn(
-                "The 'status' parameter of DatabricksError is deprecated and will be removed in a future version."
-            )
-
-        # API 1.2-specific parameters are deprecated.
-        if error:
-            warnings.warn(
-                "The 'error' parameter of DatabricksError is deprecated and will be removed in a future version."
-            )
-
-        # Retry-after is deprecated.
-        if retry_after_secs:
-            warnings.warn(
-                "The 'retry_after_secs' parameter of DatabricksError is deprecated and will be removed in a future version."
-            )
 
         if detail:
             # Handle SCIM error message details
@@ -114,12 +88,9 @@ class DatabricksError(IOError):
                 self.details.append(ErrorDetail.from_dict(d))
 
     def get_error_info(self) -> List[ErrorDetail]:
-        return self._get_details_by_type(errdetails._ERROR_INFO_TYPE)
-
-    def _get_details_by_type(self, error_type) -> List[ErrorDetail]:
         if self.details is None:
             return []
-        return [detail for detail in self.details if detail.type == error_type]
+        return [detail for detail in self.details if detail.type == errdetails._ERROR_INFO_TYPE]
 
     def get_error_details(self) -> errdetails.ErrorDetails:
         return self._error_details

databricks/sdk/service/apps.py

@@ -655,6 +655,8 @@ class AppResource:
 
     sql_warehouse: Optional[AppResourceSqlWarehouse] = None
 
+    uc_securable: Optional[AppResourceUcSecurable] = None
+
     def as_dict(self) -> dict:
         """Serializes the AppResource into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -670,6 +672,8 @@ class AppResource:
             body["serving_endpoint"] = self.serving_endpoint.as_dict()
         if self.sql_warehouse:
             body["sql_warehouse"] = self.sql_warehouse.as_dict()
+        if self.uc_securable:
+            body["uc_securable"] = self.uc_securable.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -687,6 +691,8 @@ class AppResource:
             body["serving_endpoint"] = self.serving_endpoint
         if self.sql_warehouse:
             body["sql_warehouse"] = self.sql_warehouse
+        if self.uc_securable:
+            body["uc_securable"] = self.uc_securable
         return body
 
     @classmethod
@@ -699,6 +705,7 @@ class AppResource:
             secret=_from_dict(d, "secret", AppResourceSecret),
             serving_endpoint=_from_dict(d, "serving_endpoint", AppResourceServingEndpoint),
             sql_warehouse=_from_dict(d, "sql_warehouse", AppResourceSqlWarehouse),
+            uc_securable=_from_dict(d, "uc_securable", AppResourceUcSecurable),
         )
 
 
@@ -880,6 +887,57 @@ class AppResourceSqlWarehouseSqlWarehousePermission(Enum):
     IS_OWNER = "IS_OWNER"
 
 
+@dataclass
+class AppResourceUcSecurable:
+    securable_full_name: str
+
+    securable_type: AppResourceUcSecurableUcSecurableType
+
+    permission: AppResourceUcSecurableUcSecurablePermission
+
+    def as_dict(self) -> dict:
+        """Serializes the AppResourceUcSecurable into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.permission is not None:
+            body["permission"] = self.permission.value
+        if self.securable_full_name is not None:
+            body["securable_full_name"] = self.securable_full_name
+        if self.securable_type is not None:
+            body["securable_type"] = self.securable_type.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the AppResourceUcSecurable into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.permission is not None:
+            body["permission"] = self.permission
+        if self.securable_full_name is not None:
+            body["securable_full_name"] = self.securable_full_name
+        if self.securable_type is not None:
+            body["securable_type"] = self.securable_type
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> AppResourceUcSecurable:
+        """Deserializes the AppResourceUcSecurable from a dictionary."""
+        return cls(
+            permission=_enum(d, "permission", AppResourceUcSecurableUcSecurablePermission),
+            securable_full_name=d.get("securable_full_name", None),
+            securable_type=_enum(d, "securable_type", AppResourceUcSecurableUcSecurableType),
+        )
+
+
+class AppResourceUcSecurableUcSecurablePermission(Enum):
+
+    READ_VOLUME = "READ_VOLUME"
+    WRITE_VOLUME = "WRITE_VOLUME"
+
+
+class AppResourceUcSecurableUcSecurableType(Enum):
+
+    VOLUME = "VOLUME"
+
+
 class ApplicationState(Enum):
 
     CRASHED = "CRASHED"