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

databricks/sdk/__init__.py

@@ -293,6 +293,7 @@ class WorkspaceClient:
         self._feature_engineering = pkg_ml.FeatureEngineeringAPI(self._api_client)
         self._feature_store = pkg_ml.FeatureStoreAPI(self._api_client)
         self._files = _make_files_client(self._api_client, self._config)
+        self._forecasting = pkg_ml.ForecastingAPI(self._api_client)
         self._functions = pkg_catalog.FunctionsAPI(self._api_client)
         self._genie = pkg_dashboards.GenieAPI(self._api_client)
         self._git_credentials = pkg_workspace.GitCredentialsAPI(self._api_client)
@@ -375,9 +376,8 @@ class WorkspaceClient:
         self._workspace = WorkspaceExt(self._api_client)
         self._workspace_bindings = pkg_catalog.WorkspaceBindingsAPI(self._api_client)
         self._workspace_conf = pkg_settings.WorkspaceConfAPI(self._api_client)
-        self._workspace_settings_v2 = pkg_settingsv2.WorkspaceSettingsV2API(self._api_client)
-        self._forecasting = pkg_ml.ForecastingAPI(self._api_client)
         self._workspace_iam_v2 = pkg_iamv2.WorkspaceIamV2API(self._api_client)
+        self._workspace_settings_v2 = pkg_settingsv2.WorkspaceSettingsV2API(self._api_client)
         self._groups = pkg_iam.GroupsAPI(self._api_client)
         self._service_principals = pkg_iam.ServicePrincipalsAPI(self._api_client)
         self._users = pkg_iam.UsersAPI(self._api_client)
@@ -599,6 +599,11 @@ class WorkspaceClient:
         """A feature store is a centralized repository that enables data scientists to find and share features."""
         return self._feature_store
 
+    @property
+    def forecasting(self) -> pkg_ml.ForecastingAPI:
+        """The Forecasting API allows you to create and get serverless forecasting experiments."""
+        return self._forecasting
+
     @property
     def functions(self) -> pkg_catalog.FunctionsAPI:
         """Functions implement User-Defined Functions (UDFs) in Unity Catalog."""
@@ -974,21 +979,16 @@ class WorkspaceClient:
         """This API allows updating known workspace settings for advanced users."""
         return self._workspace_conf
 
-    @property
-    def workspace_settings_v2(self) -> pkg_settingsv2.WorkspaceSettingsV2API:
-        """APIs to manage workspace level settings."""
-        return self._workspace_settings_v2
-
-    @property
-    def forecasting(self) -> pkg_ml.ForecastingAPI:
-        """The Forecasting API allows you to create and get serverless forecasting experiments."""
-        return self._forecasting
-
     @property
     def workspace_iam_v2(self) -> pkg_iamv2.WorkspaceIamV2API:
         """These APIs are used to manage identities and the workspace access of these identities in <Databricks>."""
         return self._workspace_iam_v2
 
+    @property
+    def workspace_settings_v2(self) -> pkg_settingsv2.WorkspaceSettingsV2API:
+        """APIs to manage workspace level settings."""
+        return self._workspace_settings_v2
+
     @property
     def groups(self) -> pkg_iam.GroupsAPI:
         """Groups simplify identity management, making it easier to assign access to Databricks workspace, data, and other securable objects."""
@@ -1086,11 +1086,13 @@ class AccountClient:
         self._access_control = pkg_iam.AccountAccessControlAPI(self._api_client)
         self._billable_usage = pkg_billing.BillableUsageAPI(self._api_client)
         self._budget_policy = pkg_billing.BudgetPolicyAPI(self._api_client)
+        self._budgets = pkg_billing.BudgetsAPI(self._api_client)
         self._credentials = pkg_provisioning.CredentialsAPI(self._api_client)
         self._custom_app_integration = pkg_oauth2.CustomAppIntegrationAPI(self._api_client)
         self._encryption_keys = pkg_provisioning.EncryptionKeysAPI(self._api_client)
         self._federation_policy = pkg_oauth2.AccountFederationPolicyAPI(self._api_client)
         self._groups_v2 = pkg_iam.AccountGroupsV2API(self._api_client)
+        self._iam_v2 = pkg_iamv2.AccountIamV2API(self._api_client)
         self._ip_access_lists = pkg_settings.AccountIpAccessListsAPI(self._api_client)
         self._log_delivery = pkg_billing.LogDeliveryAPI(self._api_client)
         self._metastore_assignments = pkg_catalog.AccountMetastoreAssignmentsAPI(self._api_client)
@@ -1114,8 +1116,6 @@ class AccountClient:
         self._workspace_assignment = pkg_iam.WorkspaceAssignmentAPI(self._api_client)
         self._workspace_network_configuration = pkg_settings.WorkspaceNetworkConfigurationAPI(self._api_client)
         self._workspaces = pkg_provisioning.WorkspacesAPI(self._api_client)
-        self._iam_v2 = pkg_iamv2.AccountIamV2API(self._api_client)
-        self._budgets = pkg_billing.BudgetsAPI(self._api_client)
         self._groups = pkg_iam.AccountGroupsAPI(self._api_client)
         self._service_principals = pkg_iam.AccountServicePrincipalsAPI(self._api_client)
         self._users = pkg_iam.AccountUsersAPI(self._api_client)
@@ -1143,6 +1143,11 @@ class AccountClient:
         """A service serves REST API about Budget policies."""
         return self._budget_policy
 
+    @property
+    def budgets(self) -> pkg_billing.BudgetsAPI:
+        """These APIs manage budget configurations for this account."""
+        return self._budgets
+
     @property
     def credentials(self) -> pkg_provisioning.CredentialsAPI:
         """These APIs manage credential configurations for this workspace."""
@@ -1168,6 +1173,11 @@ class AccountClient:
         """Groups simplify identity management, making it easier to assign access to Databricks account, data, and other securable objects."""
         return self._groups_v2
 
+    @property
+    def iam_v2(self) -> pkg_iamv2.AccountIamV2API:
+        """These APIs are used to manage identities and the workspace access of these identities in <Databricks>."""
+        return self._iam_v2
+
     @property
     def ip_access_lists(self) -> pkg_settings.AccountIpAccessListsAPI:
         """The Accounts IP Access List API enables account admins to configure IP access lists for access to the account console."""
@@ -1283,16 +1293,6 @@ class AccountClient:
         """These APIs manage workspaces for this account."""
         return self._workspaces
 
-    @property
-    def iam_v2(self) -> pkg_iamv2.AccountIamV2API:
-        """These APIs are used to manage identities and the workspace access of these identities in <Databricks>."""
-        return self._iam_v2
-
-    @property
-    def budgets(self) -> pkg_billing.BudgetsAPI:
-        """These APIs manage budget configurations for this account."""
-        return self._budgets
-
     @property
     def groups(self) -> pkg_iam.AccountGroupsAPI:
         """Groups simplify identity management, making it easier to assign access to Databricks account, data, and other securable objects."""

databricks/sdk/dbutils.py

@@ -281,10 +281,17 @@ def get_local_notebook_path():
     return value
 
 
+def not_supported_method_err_msg(methodName):
+    return f"Method '{methodName}' is not supported in the SDK version of DBUtils"
+
+
 class _OverrideProxyUtil:
 
     @classmethod
     def new(cls, path: str):
+        if path in cls.not_supported_override_paths:
+            raise ValueError(cls.not_supported_override_paths[path])
+
         if len(cls.__get_matching_overrides(path)) > 0:
             return _OverrideProxyUtil(path)
         return None
@@ -301,6 +308,16 @@ class _OverrideProxyUtil:
         "notebook.entry_point.getDbutils().notebook().getContext().notebookPath().get()": get_local_notebook_path,
     }
 
+    # These paths work the same as 'proxy_override_paths' but instead of using a local implementation we raise an exception.
+    not_supported_override_paths = {
+        # The object returned by 'credentials.getServiceCredentialProvider()' can't be serialized to JSON.
+        # Without this override, the command would fail with an error 'TypeError: Object of type Session is not JSON serializable'.
+        # We override it to show a better error message
+        "credentials.getServiceCredentialsProvider": not_supported_method_err_msg(
+            "credentials.getServiceCredentialsProvider"
+        ),
+    }
+
     @classmethod
     def __get_matching_overrides(cls, path: str):
         return [x for x in cls.proxy_override_paths.keys() if x.startswith(path)]

databricks/sdk/mixins/files.py

@@ -1064,7 +1064,7 @@ class FilesExt(files.FilesAPI):
     def upload(
         self,
         file_path: str,
-        content: BinaryIO,
+        contents: BinaryIO,
         *,
         overwrite: Optional[bool] = None,
         part_size: Optional[int] = None,
@@ -1076,7 +1076,7 @@ class FilesExt(files.FilesAPI):
 
         :param file_path: str
             The absolute remote path of the target file, e.g. /Volumes/path/to/your/file
-        :param content: BinaryIO
+        :param contents: BinaryIO
             The contents of the file to upload. This must be a BinaryIO stream.
         :param overwrite: bool (optional)
             If true, an existing file will be overwritten. When not specified, assumed True.
@@ -1096,7 +1096,7 @@ class FilesExt(files.FilesAPI):
 
         if self._config.disable_experimental_files_api_client:
             _LOG.info("Disable experimental files API client, will use the original upload method.")
-            super().upload(file_path=file_path, contents=content, overwrite=overwrite)
+            super().upload(file_path=file_path, contents=contents, overwrite=overwrite)
             return UploadStreamResult()
 
         _LOG.debug(f"Uploading file from BinaryIO stream")
@@ -1107,12 +1107,12 @@ class FilesExt(files.FilesAPI):
 
         # Determine content length if the stream is seekable
         content_length = None
-        if content.seekable():
+        if contents.seekable():
             _LOG.debug(f"Uploading using seekable mode")
             # If the stream is seekable, we can read its size.
-            content.seek(0, os.SEEK_END)
-            content_length = content.tell()
-            content.seek(0)
+            contents.seek(0, os.SEEK_END)
+            content_length = contents.tell()
+            contents.seek(0)
 
         # Get optimized part size and batch size based on content length and provided part size
         optimized_part_size, optimized_batch_size = self._get_optimized_performance_parameters_for_upload(
@@ -1135,17 +1135,17 @@ class FilesExt(files.FilesAPI):
         )
 
         if ctx.use_parallel:
-            self._parallel_upload_from_stream(ctx, content)
+            self._parallel_upload_from_stream(ctx, contents)
             return UploadStreamResult()
         elif ctx.content_length is not None:
-            self._upload_single_thread_with_known_size(ctx, content)
+            self._upload_single_thread_with_known_size(ctx, contents)
             return UploadStreamResult()
         else:
             _LOG.debug(f"Uploading using non-seekable mode")
             # If the stream is not seekable, we cannot determine its size.
             # We will use a multipart upload.
             _LOG.debug(f"Using multipart upload for non-seekable input stream of unknown size for file {file_path}")
-            self._single_thread_multipart_upload(ctx, content)
+            self._single_thread_multipart_upload(ctx, contents)
             return UploadStreamResult()
 
     def upload_from(

databricks/sdk/service/agentbricks.py

@@ -238,6 +238,7 @@ class AgentBricksAPI:
 
         :returns: :class:`CustomLlm`
         """
+
         body = {}
         if agent_artifact_path is not None:
             body["agent_artifact_path"] = agent_artifact_path
@@ -328,6 +329,7 @@ class AgentBricksAPI:
 
         :returns: :class:`CustomLlm`
         """
+
         body = {}
         if custom_llm is not None:
             body["custom_llm"] = custom_llm.as_dict()

databricks/sdk/service/apps.py

@@ -790,11 +790,13 @@ class AppManifestAppResourceUcSecurableSpecUcSecurablePermission(Enum):
 
     MANAGE = "MANAGE"
     READ_VOLUME = "READ_VOLUME"
+    SELECT = "SELECT"
     WRITE_VOLUME = "WRITE_VOLUME"
 
 
 class AppManifestAppResourceUcSecurableSpecUcSecurableType(Enum):
 
+    TABLE = "TABLE"
     VOLUME = "VOLUME"
 
 
@@ -1867,6 +1869,7 @@ class AppsAPI:
           Long-running operation waiter for :class:`App`.
           See :method:wait_get_app_active for more details.
         """
+
         body = app.as_dict()
         query = {}
         if no_compute is not None:
@@ -1903,6 +1906,7 @@ class AppsAPI:
           Long-running operation waiter for :class:`AppUpdate`.
           See :method:wait_get_update_app_succeeded for more details.
         """
+
         body = {}
         if app is not None:
             body["app"] = app.as_dict()
@@ -1949,6 +1953,7 @@ class AppsAPI:
           Long-running operation waiter for :class:`AppDeployment`.
           See :method:wait_get_deployment_app_succeeded for more details.
         """
+
         body = app_deployment.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2124,6 +2129,7 @@ class AppsAPI:
 
         :returns: :class:`AppPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -2189,6 +2195,7 @@ class AppsAPI:
 
         :returns: :class:`App`
         """
+
         body = app.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2209,6 +2216,7 @@ class AppsAPI:
 
         :returns: :class:`AppPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -2234,6 +2242,7 @@ class AppsSettingsAPI:
 
         :returns: :class:`CustomTemplate`
         """
+
         body = template.as_dict()
         headers = {
             "Accept": "application/json",
@@ -2316,6 +2325,7 @@ class AppsSettingsAPI:
 
         :returns: :class:`CustomTemplate`
         """
+
         body = template.as_dict()
         headers = {
             "Accept": "application/json",

databricks/sdk/service/billing.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import logging
+import uuid
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any, BinaryIO, Dict, Iterator, List, Optional
@@ -1067,9 +1068,6 @@ class LogDeliveryConfiguration:
     [Configuring audit logs]: https://docs.databricks.com/administration-guide/account-settings/audit-logs.html
     [View billable usage]: https://docs.databricks.com/administration-guide/account-settings/usage.html"""
 
-    account_id: str
-    """Databricks account ID."""
-
     credentials_id: str
     """The ID for a method:credentials/create that represents the AWS IAM role with policy and trust
     relationship as described in the main billable usage documentation page. See [Configure billable
@@ -1083,6 +1081,9 @@ class LogDeliveryConfiguration:
     
     [Configure billable usage delivery]: https://docs.databricks.com/administration-guide/account-settings/billable-usage-delivery.html"""
 
+    account_id: Optional[str] = None
+    """Databricks account ID."""
+
     config_id: Optional[str] = None
     """The unique UUID of log delivery configuration"""
 
@@ -1565,6 +1566,9 @@ class BudgetPolicyAPI:
 
         :returns: :class:`BudgetPolicy`
         """
+
+        if request_id is None or request_id == "":
+            request_id = str(uuid.uuid4())
         body = {}
         if policy is not None:
             body["policy"] = policy.as_dict()
@@ -1679,6 +1683,7 @@ class BudgetPolicyAPI:
 
         :returns: :class:`BudgetPolicy`
         """
+
         body = policy.as_dict()
         query = {}
         if limit_config is not None:
@@ -1715,6 +1720,7 @@ class BudgetsAPI:
 
         :returns: :class:`CreateBudgetConfigurationResponse`
         """
+
         body = {}
         if budget is not None:
             body["budget"] = budget.as_dict()
@@ -1797,6 +1803,7 @@ class BudgetsAPI:
 
         :returns: :class:`UpdateBudgetConfigurationResponse`
         """
+
         body = {}
         if budget is not None:
             body["budget"] = budget.as_dict()
@@ -1895,6 +1902,7 @@ class LogDeliveryAPI:
 
         :returns: :class:`WrappedLogDeliveryConfiguration`
         """
+
         body = {}
         if log_delivery_configuration is not None:
             body["log_delivery_configuration"] = log_delivery_configuration.as_dict()
@@ -1989,6 +1997,7 @@ class LogDeliveryAPI:
 
 
         """
+
         body = {}
         if status is not None:
             body["status"] = status.value
@@ -2026,6 +2035,7 @@ class UsageDashboardsAPI:
 
         :returns: :class:`CreateBillingUsageDashboardResponse`
         """
+
         body = {}
         if dashboard_type is not None:
             body["dashboard_type"] = dashboard_type.value