Merge branch 'main' into fix-indefinite-retry

Author: parthban-db

.codegen/_openapi_sha

@@ -1 +1 @@
-ed8b5e7ec1f143395503eebae9fbc74cf8c309bc
+9c14189832570a8bb6f55441db02b49f04c15a1d

.github/workflows/tagging.yml

@@ -21,11 +21,13 @@ jobs:
     # Only run the tag job if the trigger is manual (workflow_dispatch) or
     # the repository has been approved for automated releases.
     #
-    # To disable release for a repository, simply exclude it from the if 
+    # To disable release for a repository, simply exclude it from the if
     # condition.
     if: >-
       github.event_name == 'workflow_dispatch' ||
-      github.repository == 'databricks/databricks-sdk-go'
+      github.repository == 'databricks/databricks-sdk-go' ||
+      github.repository == 'databricks/databricks-sdk-py' ||
+      github.repository == 'databricks/databricks-sdk-java'
     environment: "release-is"
     runs-on:
       group: databricks-deco-testing-runner-group

.release_metadata.json

@@ -1,3 +1,3 @@
 {
-    "timestamp": "2026-01-15 07:46:58+0000"
+    "timestamp": "2026-01-22 20:29:25+0000"
 }

CHANGELOG.md

@@ -1,5 +1,63 @@
 # Version changelog
 
+## Release v0.80.0 (2026-01-22)
+
+### Internal Changes
+
+* Support CLI Profiles for Unified Host.
+
+
+## Release v0.79.0 (2026-01-22)
+
+### API Changes
+* Add `full_refresh_window` field for `databricks.sdk.service.pipelines.IngestionPipelineDefinition`.
+* Add `auto_full_refresh_policy` field for `databricks.sdk.service.pipelines.TableSpecificConfig`.
+* Add `hosts` field for `databricks.sdk.service.postgres.EndpointStatus`.
+* Add `owner` field for `databricks.sdk.service.postgres.ProjectStatus`.
+* Add `validity_check_configurations` field for `databricks.sdk.service.qualitymonitorv2.QualityMonitor`.
+* Add `burst_scaling_enabled` field for `databricks.sdk.service.serving.PtServedModel`.
+* Add `system_managed` enum value for `databricks.sdk.service.jobs.JobDeploymentKind`.
+* Add `endpoint_type_read_write` and `endpoint_type_read_only` enum values for `databricks.sdk.service.postgres.EndpointType`.
+* Add `deleted` enum value for `databricks.sdk.service.vectorsearch.EndpointStatusState`.
+* [Breaking] Remove `host`, `last_active_time`, `start_time` and `suspend_time` fields for `databricks.sdk.service.postgres.EndpointStatus`.
+* [Breaking] Remove `compute_last_active_time` field for `databricks.sdk.service.postgres.ProjectStatus`.
+* [Breaking] Remove `read_write` and `read_only` enum values for `databricks.sdk.service.postgres.EndpointType`.
+* Add `create_default_warehouse_override()`, `delete_default_warehouse_override()`, `get_default_warehouse_override()`, `list_default_warehouse_overrides()` and `update_default_warehouse_override()` methods for [w.warehouses](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/sql/warehouses.html) workspace-level service.
+* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.ClusterAttributes`.
+* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.ClusterDetails`.
+* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.ClusterSpec`.
+* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.CreateCluster`.
+* Add `node_type_flexibility` field for `databricks.sdk.service.compute.CreateInstancePool`.
+* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.EditCluster`.
+* Add `node_type_flexibility` field for `databricks.sdk.service.compute.EditInstancePool`.
+* Add `node_type_flexibility` field for `databricks.sdk.service.compute.GetInstancePool`.
+* Add `node_type_flexibility` field for `databricks.sdk.service.compute.InstancePoolAndStats`.
+* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.UpdateClusterResource`.
+* Add `expire_time` and `ttl` fields for `databricks.sdk.service.postgres.BranchSpec`.
+* Add `expire_time` field for `databricks.sdk.service.postgres.BranchStatus`.
+* [Breaking] Change `create_branch()`, `create_endpoint()` and `create_project()` methods for [w.postgres](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/postgres/postgres.html) workspace-level service with new required argument order.
+* [Breaking] Change `branch_id` field for `databricks.sdk.service.postgres.CreateBranchRequest` to be required.
+* [Breaking] Change `endpoint_id` field for `databricks.sdk.service.postgres.CreateEndpointRequest` to be required.
+* [Breaking] Change `project_id` field for `databricks.sdk.service.postgres.CreateProjectRequest` to be required.
+* [Breaking] Remove `default` field for `databricks.sdk.service.postgres.BranchSpec`.
+* [Breaking] Remove `settings` field for `databricks.sdk.service.postgres.ProjectSpec`.
+* [Breaking] Remove `settings` field for `databricks.sdk.service.postgres.ProjectStatus`.
+* Add `generate_database_credential()` method for [w.postgres](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/postgres/postgres.html) workspace-level service.
+* Add `get_public_account_user_preference()`, `list_account_user_preferences_metadata()` and `patch_public_account_user_preference()` methods for [a.account_settings_v2](https://databricks-sdk-py.readthedocs.io/en/latest/account/settingsv2/account_settings_v2.html) account-level service.
+* Add `stderr` field for `databricks.sdk.service.compute.InitScriptInfoAndExecutionDetails`.
+* Add `no_expiry` field for `databricks.sdk.service.postgres.BranchSpec`.
+* Add `outputs` field for `databricks.sdk.service.serving.QueryEndpointResponse`.
+* Add `session_id` field for `databricks.sdk.service.sql.QueryInfo`.
+* Add `principal_id` field for `databricks.sdk.service.workspace.CreateCredentialsRequest`.
+* Add `principal_id` field for `databricks.sdk.service.workspace.DeleteCredentialsRequest`.
+* Add `principal_id` field for `databricks.sdk.service.workspace.GetCredentialsRequest`.
+* Add `principal_id` field for `databricks.sdk.service.workspace.ListCredentialsRequest`.
+* Add `principal_id` field for `databricks.sdk.service.workspace.UpdateCredentialsRequest`.
+* Add `force` field for `databricks.sdk.service.pipelines.DeletePipelineRequest`.
+* Add `no_suspension` field for `databricks.sdk.service.postgres.EndpointSpec`.
+* Add `no_suspension` field for `databricks.sdk.service.postgres.ProjectDefaultEndpointSettings`.
+
+
 ## Release v0.78.0 (2026-01-15)
 
 ### New Features and Improvements

NEXT_CHANGELOG.md

@@ -1,6 +1,6 @@
 # NEXT CHANGELOG
 
-## Release v0.79.0
+## Release v0.81.0
 
 ### New Features and Improvements
 
@@ -17,46 +17,6 @@
 ### Internal Changes
 
 ### API Changes
-* Add `full_refresh_window` field for `databricks.sdk.service.pipelines.IngestionPipelineDefinition`.
-* Add `auto_full_refresh_policy` field for `databricks.sdk.service.pipelines.TableSpecificConfig`.
-* Add `hosts` field for `databricks.sdk.service.postgres.EndpointStatus`.
-* Add `owner` field for `databricks.sdk.service.postgres.ProjectStatus`.
-* Add `validity_check_configurations` field for `databricks.sdk.service.qualitymonitorv2.QualityMonitor`.
-* Add `burst_scaling_enabled` field for `databricks.sdk.service.serving.PtServedModel`.
-* Add `system_managed` enum value for `databricks.sdk.service.jobs.JobDeploymentKind`.
-* Add `endpoint_type_read_write` and `endpoint_type_read_only` enum values for `databricks.sdk.service.postgres.EndpointType`.
-* Add `deleted` enum value for `databricks.sdk.service.vectorsearch.EndpointStatusState`.
-* [Breaking] Remove `host`, `last_active_time`, `start_time` and `suspend_time` fields for `databricks.sdk.service.postgres.EndpointStatus`.
-* [Breaking] Remove `compute_last_active_time` field for `databricks.sdk.service.postgres.ProjectStatus`.
-* [Breaking] Remove `read_write` and `read_only` enum values for `databricks.sdk.service.postgres.EndpointType`.
-* Add `create_default_warehouse_override()`, `delete_default_warehouse_override()`, `get_default_warehouse_override()`, `list_default_warehouse_overrides()` and `update_default_warehouse_override()` methods for [w.warehouses](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/sql/warehouses.html) workspace-level service.
-* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.ClusterAttributes`.
-* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.ClusterDetails`.
-* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.ClusterSpec`.
-* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.CreateCluster`.
-* Add `node_type_flexibility` field for `databricks.sdk.service.compute.CreateInstancePool`.
-* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.EditCluster`.
-* Add `node_type_flexibility` field for `databricks.sdk.service.compute.EditInstancePool`.
-* Add `node_type_flexibility` field for `databricks.sdk.service.compute.GetInstancePool`.
-* Add `node_type_flexibility` field for `databricks.sdk.service.compute.InstancePoolAndStats`.
-* Add `driver_node_type_flexibility` and `worker_node_type_flexibility` fields for `databricks.sdk.service.compute.UpdateClusterResource`.
-* Add `expire_time` and `ttl` fields for `databricks.sdk.service.postgres.BranchSpec`.
-* Add `expire_time` field for `databricks.sdk.service.postgres.BranchStatus`.
-* [Breaking] Change `create_branch()`, `create_endpoint()` and `create_project()` methods for [w.postgres](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/postgres/postgres.html) workspace-level service with new required argument order.
-* [Breaking] Change `branch_id` field for `databricks.sdk.service.postgres.CreateBranchRequest` to be required.
-* [Breaking] Change `endpoint_id` field for `databricks.sdk.service.postgres.CreateEndpointRequest` to be required.
-* [Breaking] Change `project_id` field for `databricks.sdk.service.postgres.CreateProjectRequest` to be required.
-* [Breaking] Remove `default` field for `databricks.sdk.service.postgres.BranchSpec`.
-* [Breaking] Remove `settings` field for `databricks.sdk.service.postgres.ProjectSpec`.
-* [Breaking] Remove `settings` field for `databricks.sdk.service.postgres.ProjectStatus`.
-* Add `generate_database_credential()` method for [w.postgres](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/postgres/postgres.html) workspace-level service.
-* Add `get_public_account_user_preference()`, `list_account_user_preferences_metadata()` and `patch_public_account_user_preference()` methods for [a.account_settings_v2](https://databricks-sdk-py.readthedocs.io/en/latest/account/settingsv2/account_settings_v2.html) account-level service.
-* Add `stderr` field for `databricks.sdk.service.compute.InitScriptInfoAndExecutionDetails`.
-* Add `no_expiry` field for `databricks.sdk.service.postgres.BranchSpec`.
-* Add `outputs` field for `databricks.sdk.service.serving.QueryEndpointResponse`.
-* Add `session_id` field for `databricks.sdk.service.sql.QueryInfo`.
-* Add `principal_id` field for `databricks.sdk.service.workspace.CreateCredentialsRequest`.
-* Add `principal_id` field for `databricks.sdk.service.workspace.DeleteCredentialsRequest`.
-* Add `principal_id` field for `databricks.sdk.service.workspace.GetCredentialsRequest`.
-* Add `principal_id` field for `databricks.sdk.service.workspace.ListCredentialsRequest`.
-* Add `principal_id` field for `databricks.sdk.service.workspace.UpdateCredentialsRequest`.
+* Add `postgres_role` field for `databricks.sdk.service.postgres.RoleRoleSpec`.
+* Add `postgres_role` field for `databricks.sdk.service.postgres.RoleRoleStatus`.
+* Add `purge` field for `databricks.sdk.service.sql.TrashAlertV2Request`.

databricks/sdk/client_types.py

@@ -0,0 +1,16 @@
+from enum import Enum
+
+
+class HostType(Enum):
+    """Enum representing the type of Databricks host."""
+
+    ACCOUNTS = "accounts"
+    WORKSPACE = "workspace"
+    UNIFIED = "unified"
+
+
+class ClientType(Enum):
+    """Enum representing the type of client configuration."""
+
+    ACCOUNT = "account"
+    WORKSPACE = "workspace"

databricks/sdk/config.py

@@ -7,13 +7,13 @@
 import re
 import sys
 import urllib.parse
-from enum import Enum
 from typing import Dict, Iterable, List, Optional
 
 import requests
 
 from . import useragent
 from ._base_client import _fix_host_if_needed
+from .client_types import ClientType, HostType
 from .clock import Clock, RealClock
 from .credentials_provider import (CredentialsStrategy, DefaultCredentials,
                                    OAuthCredentialsProvider)
@@ -26,21 +26,6 @@
 logger = logging.getLogger("databricks.sdk")
 
 
-class HostType(Enum):
-    """Enum representing the type of Databricks host."""
-
-    ACCOUNTS = "accounts"
-    WORKSPACE = "workspace"
-    UNIFIED = "unified"
-
-
-class ClientType(Enum):
-    """Enum representing the type of client configuration."""
-
-    ACCOUNT = "account"
-    WORKSPACE = "workspace"
-
-
 class ConfigAttribute:
     """Configuration attribute metadata and descriptor protocols."""
 

databricks/sdk/credentials_provider.py

@@ -21,6 +21,7 @@
 from google.oauth2 import service_account  # type: ignore
 
 from . import azure, oauth, oidc, oidc_token_supplier
+from .client_types import ClientType
 
 CredentialsProvider = Callable[[], Dict[str, str]]
 
@@ -430,9 +431,9 @@ def _oidc_credentials_provider(
 
     # Determine the audience for token exchange
     audience = cfg.token_audience
-    if audience is None and cfg.is_account_client:
+    if audience is None and cfg.client_type == ClientType.ACCOUNT:
         audience = cfg.account_id
-    if audience is None and not cfg.is_account_client:
+    if audience is None and cfg.client_type != ClientType.ACCOUNT:
         audience = cfg.oidc_endpoints.token_endpoint
 
     # Try to get an OIDC token. If no supplier returns a token, we cannot use this authentication mode.
@@ -589,7 +590,7 @@ def token() -> oauth.Token:
     def refreshed_headers() -> Dict[str, str]:
         credentials.refresh(request)
         headers = {"Authorization": f"Bearer {credentials.token}"}
-        if cfg.is_account_client:
+        if cfg.client_type == ClientType.ACCOUNT:
             gcp_credentials.refresh(request)
             headers["X-Databricks-GCP-SA-Access-Token"] = gcp_credentials.token
         return headers
@@ -630,7 +631,7 @@ def token() -> oauth.Token:
     def refreshed_headers() -> Dict[str, str]:
         id_creds.refresh(request)
         headers = {"Authorization": f"Bearer {id_creds.token}"}
-        if cfg.is_account_client:
+        if cfg.client_type == ClientType.ACCOUNT:
             gcp_impersonated_credentials.refresh(request)
             headers["X-Databricks-GCP-SA-Access-Token"] = gcp_impersonated_credentials.token
         return headers
@@ -852,7 +853,14 @@ class DatabricksCliTokenSource(CliTokenSource):
 
     def __init__(self, cfg: "Config"):
         args = ["auth", "token", "--host", cfg.host]
-        if cfg.is_account_client:
+        if cfg.experimental_is_unified_host:
+            # For unified hosts, pass account_id, workspace_id, and experimental flag
+            args += ["--experimental-is-unified-host"]
+            if cfg.account_id:
+                args += ["--account-id", cfg.account_id]
+            if cfg.workspace_id:
+                args += ["--workspace-id", str(cfg.workspace_id)]
+        elif cfg.client_type == ClientType.ACCOUNT:
             args += ["--account-id", cfg.account_id]
 
         cli_path = cfg.databricks_cli_path