databricks-sdk 0.60.0__py3-none-any.whl → 0.62.0__py3-none-any.whl

databricks/sdk/__init__.py

@@ -59,7 +59,9 @@ from databricks.sdk.service.catalog import (AccountMetastoreAssignmentsAPI,
                                             TableConstraintsAPI, TablesAPI,
                                             TemporaryTableCredentialsAPI,
                                             VolumesAPI, WorkspaceBindingsAPI)
-from databricks.sdk.service.cleanrooms import (CleanRoomAssetsAPI,
+from databricks.sdk.service.cleanrooms import (CleanRoomAssetRevisionsAPI,
+                                               CleanRoomAssetsAPI,
+                                               CleanRoomAutoApprovalRulesAPI,
                                                CleanRoomsAPI,
                                                CleanRoomTaskRunsAPI)
 from databricks.sdk.service.compute import (ClusterPoliciesAPI, ClustersAPI,
@@ -247,7 +249,9 @@ class WorkspaceClient:
         self._apps = pkg_apps.AppsAPI(self._api_client)
         self._artifact_allowlists = pkg_catalog.ArtifactAllowlistsAPI(self._api_client)
         self._catalogs = pkg_catalog.CatalogsAPI(self._api_client)
+        self._clean_room_asset_revisions = pkg_cleanrooms.CleanRoomAssetRevisionsAPI(self._api_client)
         self._clean_room_assets = pkg_cleanrooms.CleanRoomAssetsAPI(self._api_client)
+        self._clean_room_auto_approval_rules = pkg_cleanrooms.CleanRoomAutoApprovalRulesAPI(self._api_client)
         self._clean_room_task_runs = pkg_cleanrooms.CleanRoomTaskRunsAPI(self._api_client)
         self._clean_rooms = pkg_cleanrooms.CleanRoomsAPI(self._api_client)
         self._cluster_policies = pkg_compute.ClusterPoliciesAPI(self._api_client)
@@ -411,11 +415,21 @@ class WorkspaceClient:
         """A catalog is the first layer of Unity Catalog’s three-level namespace."""
         return self._catalogs
 
+    @property
+    def clean_room_asset_revisions(self) -> pkg_cleanrooms.CleanRoomAssetRevisionsAPI:
+        """Clean Room Asset Revisions denote new versions of uploaded assets (e.g."""
+        return self._clean_room_asset_revisions
+
     @property
     def clean_room_assets(self) -> pkg_cleanrooms.CleanRoomAssetsAPI:
         """Clean room assets are data and code objects — Tables, volumes, and notebooks that are shared with the clean room."""
         return self._clean_room_assets
 
+    @property
+    def clean_room_auto_approval_rules(self) -> pkg_cleanrooms.CleanRoomAutoApprovalRulesAPI:
+        """Clean room auto-approval rules automatically create an approval on your behalf when an asset (e.g."""
+        return self._clean_room_auto_approval_rules
+
     @property
     def clean_room_task_runs(self) -> pkg_cleanrooms.CleanRoomTaskRunsAPI:
         """Clean room task runs are the executions of notebooks in a clean room."""

databricks/sdk/config.py

@@ -13,7 +13,8 @@ import requests
 from . import useragent
 from ._base_client import _fix_host_if_needed
 from .clock import Clock, RealClock
-from .credentials_provider import CredentialsStrategy, DefaultCredentials
+from .credentials_provider import (CredentialsStrategy, DefaultCredentials,
+                                   OAuthCredentialsProvider)
 from .environments import (ALL_ENVS, AzureEnvironment, Cloud,
                            DatabricksEnvironment, get_environment_for_hostname)
 from .oauth import (OidcEndpoints, Token, get_account_endpoints,
@@ -200,7 +201,19 @@ class Config:
             raise ValueError(message) from e
 
     def oauth_token(self) -> Token:
-        return self._credentials_strategy.oauth_token(self)
+        """Returns the OAuth token from the current credential provider.
+
+        This method only works when using OAuth-based authentication methods.
+        If the current credential provider is an OAuthCredentialsProvider, it reuses
+        the existing provider. Otherwise, it raises a ValueError indicating that
+        OAuth tokens are not available for the current authentication method.
+        """
+        if isinstance(self._header_factory, OAuthCredentialsProvider):
+            return self._header_factory.oauth_token()
+        raise ValueError(
+            f"OAuth tokens are not available for {self.auth_type} authentication. "
+            f"Use an OAuth-based authentication method to access OAuth tokens."
+        )
 
     def wrap_debug_info(self, message: str) -> str:
         debug_string = self.debug_string()

databricks/sdk/service/apps.py

@@ -32,6 +32,8 @@ class App:
     app_status: Optional[ApplicationStatus] = None
 
     budget_policy_id: Optional[str] = None
+    """TODO: Deprecate this field after serverless entitlements are released to all prod stages and the
+    new usage_policy_id is properly populated and used."""
 
     compute_status: Optional[ComputeStatus] = None
 
@@ -49,6 +51,8 @@ class App:
     """The description of the app."""
 
     effective_budget_policy_id: Optional[str] = None
+    """TODO: Deprecate this field after serverless entitlements are released to all prod stages and the
+    new usage_policy_id is properly populated and used."""
 
     effective_user_api_scopes: Optional[List[str]] = None
     """The effective api scopes granted to the user access token."""