databricks-sdk 0.153.0__tar.gz → 0.154.0__tar.gz

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/CHANGELOG.md

@@ -1,5 +1,22 @@
 # Version changelog
 
+## Release v0.154.0 (2026-10-19)
+
+### API Changes
+* Add `create_directory()`, `delete_path()`, `get_metadata()`, `list_directory_contents()` and `rename_path()` methods for [w.sandbox](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/sandbox/sandbox.html) workspace-level service.
+* Add `secrets` field for `databricks.sdk.service.catalog.ConnectionInfo`.
+* Add `secrets` field for `databricks.sdk.service.catalog.CreateConnection`.
+* Add `governed_tags` field for `databricks.sdk.service.catalog.PolicyInfo`.
+* Add `secrets` field for `databricks.sdk.service.catalog.UpdateConnection`.
+* Add `has_replication_point_blocking_errors` and `initial_replication_progress` fields for `databricks.sdk.service.disasterrecovery.FailoverGroup`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.BaseRun`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.CreateJob`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.JobSettings`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.RepairRun`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.Run`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.RunNow`.
+* Add `duration_target` field for `databricks.sdk.service.jobs.SubmitRun`.
+
 ## Release v0.153.0 (2026-10-15)
 
 ### API Changes

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-sdk
-Version: 0.153.0
+Version: 0.154.0
 Summary: Databricks SDK for Python (Beta)
 Project-URL: Documentation, https://databricks-sdk-py.readthedocs.io
 Keywords: databricks,sdk

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/databricks/sdk/__init__.py

@@ -829,7 +829,7 @@ class WorkspaceClient:
 
     @property
     def knowledge_assistants(self) -> pkg_knowledgeassistants.KnowledgeAssistantsAPI:
-        """Manage Knowledge Assistants and related resources."""
+        """Manage legacy Knowledge Assistants and related resources."""
         return self._knowledge_assistants
 
     @property
@@ -1094,7 +1094,7 @@ class WorkspaceClient:
 
     @property
     def supervisor_agents(self) -> pkg_supervisoragents.SupervisorAgentsAPI:
-        """Manage Supervisor Agents and related resources."""
+        """Manage legacy Supervisor Agents and related resources."""
         return self._supervisor_agents
 
     @property

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/databricks/sdk/service/agentkit.py

@@ -138,10 +138,10 @@ class ForkSessionResponse:
 
 @dataclass
 class ListManagedMemoryEntriesResponse:
-    """Response containing managed memory entries."""
+    """Response containing memory entries."""
 
     managed_memory_entries: Optional[List[ManagedMemoryEntry]] = None
-    """Managed memory entries matching the request and selected response view."""
+    """Memory entries matching the request and selected response view."""
 
     next_page_token: Optional[str] = None
     """Opaque pagination token. This field is omitted when there are no more results."""
@@ -175,10 +175,10 @@ class ListManagedMemoryEntriesResponse:
 
 @dataclass
 class ListManagedMemoryStoresResponse:
-    """Response containing managed memory stores in the caller's workspace."""
+    """Response containing memory stores in the caller's workspace."""
 
     managed_memory_stores: Optional[List[ManagedMemoryStore]] = None
-    """Managed memory stores in the caller's workspace."""
+    """Memory stores in the caller's workspace."""
 
     next_page_token: Optional[str] = None
     """Opaque pagination token. This field is omitted when there are no more results."""
@@ -320,7 +320,7 @@ class ListSessionsResponse:
 
 @dataclass
 class ManagedMemoryEntry:
-    """A workspace-scoped entry in a managed memory store."""
+    """A workspace-scoped entry in a memory store."""
 
     actor_id: str
     """Customer-provided identifier for the actor whose memory this entry represents."""
@@ -416,10 +416,10 @@ class ManagedMemoryEntry:
 
 @dataclass
 class ManagedMemoryEntrySearchResult:
-    """One relevance-ranked managed memory search result."""
+    """One relevance-ranked memory search result."""
 
     managed_memory_entry: Optional[ManagedMemoryEntry] = None
-    """Managed memory entry matching the query."""
+    """Memory entry matching the query."""
 
     score: Optional[float] = None
     """Relevance score for the result. Higher scores are more relevant."""
@@ -451,14 +451,14 @@ class ManagedMemoryEntrySearchResult:
 
 
 class ManagedMemoryEntrySourceType(Enum):
-    """Identifies the source that created a managed memory entry."""
+    """Identifies the source that created a memory entry."""
 
     MANAGED_MEMORY_ENTRY_SOURCE_TYPE_AGENT = "MANAGED_MEMORY_ENTRY_SOURCE_TYPE_AGENT"
     MANAGED_MEMORY_ENTRY_SOURCE_TYPE_DREAMER = "MANAGED_MEMORY_ENTRY_SOURCE_TYPE_DREAMER"
 
 
 class ManagedMemoryEntryView(Enum):
-    """Controls how much managed memory entry data is returned."""
+    """Controls how much memory entry data is returned."""
 
     MANAGED_MEMORY_ENTRY_VIEW_BASIC = "MANAGED_MEMORY_ENTRY_VIEW_BASIC"
     MANAGED_MEMORY_ENTRY_VIEW_FULL = "MANAGED_MEMORY_ENTRY_VIEW_FULL"
@@ -466,7 +466,7 @@ class ManagedMemoryEntryView(Enum):
 
 @dataclass
 class ManagedMemoryStore:
-    """A workspace-scoped managed memory store backed by service-managed storage."""
+    """A workspace-scoped memory store backed by service-managed storage."""
 
     create_time: Optional[Timestamp] = None
     """Time when the store was created."""
@@ -480,9 +480,9 @@ class ManagedMemoryStore:
     """Human-readable description of the memory store."""
 
     display_name: Optional[str] = None
-    """Deprecated compatibility alias for the caller-provided managed memory store ID. Canonical
-    clients provide the ID through ``CreateMemoryStoreRequest.managed_memory_store_id`` and use
-    ``name`` as the resource identifier."""
+    """Deprecated compatibility alias for the caller-provided memory store ID. Canonical clients
+    provide the ID through ``CreateMemoryStoreRequest.managed_memory_store_id`` and use ``name`` as
+    the resource identifier."""
 
     name: Optional[str] = None
     """Resource name in the form ``memory-stores/{managed_memory_store_id}``."""
@@ -591,7 +591,7 @@ class PopSessionItemResponse:
 
 @dataclass
 class SearchManagedMemoryEntriesResponse:
-    """Response containing managed memory entries ranked by relevance."""
+    """Response containing memory entries ranked by relevance."""
 
     managed_memory_entries: Optional[List[ManagedMemoryEntry]] = None
     """Deprecated compatibility alias for clients migrating to ``results``. This contains the same
@@ -852,7 +852,7 @@ class SessionStore:
 
 @dataclass
 class StorageBackend:
-    """Service-managed storage backing a managed memory store."""
+    """Service-managed storage backing a memory store."""
 
     backend_id: Optional[str] = None
     """Backend-specific identifier. For Lakebase, this is the project ID."""
@@ -885,7 +885,7 @@ class StorageBackend:
 
 
 class StorageBackendType(Enum):
-    """Type of service-managed storage backing a managed memory store."""
+    """Type of service-managed storage backing a memory store."""
 
     STORAGE_BACKEND_TYPE_LAKEBASE = "STORAGE_BACKEND_TYPE_LAKEBASE"
 
@@ -951,19 +951,18 @@ class AgentKitAPI:
     def create_memory(
         self, parent: str, managed_memory_entry: ManagedMemoryEntry, *, managed_memory_entry_id: Optional[str] = None
     ) -> ManagedMemoryEntry:
-        """Creates a managed memory entry using exclusive-create semantics. Callers may choose the entry ID; the
-        service generates one when it is omitted. Returns ``ALREADY_EXISTS`` when an entry with the same
-        actor, session, and path already exists. Omitted ``session_id`` is its own uniqueness key: two
+        """Creates a memory entry using exclusive-create semantics. Callers may choose the entry ID; the service
+        generates one when it is omitted. Returns ``ALREADY_EXISTS`` when an entry with the same actor,
+        session, and path already exists. Omitted ``session_id`` is its own uniqueness key: two
         omitted-session entries with the same actor and path conflict, but an omitted-session entry does not
         conflict with a session-scoped entry at the same actor and path.
 
         :param parent: str
-          Managed memory store that will contain the entry, in the form
-          ``memory-stores/{managed_memory_store_id}``.
+          Memory store that will contain the entry, in the form ``memory-stores/{managed_memory_store_id}``.
         :param managed_memory_entry: :class:`ManagedMemoryEntry`
-          The managed memory entry to create.
+          The memory entry to create.
         :param managed_memory_entry_id: str (optional)
-          Optional caller-selected managed memory entry ID. The service generates an ID when omitted.
+          Optional caller-selected memory entry ID. The service generates an ID when omitted.
 
         :returns: :class:`ManagedMemoryEntry`
         """
@@ -987,12 +986,12 @@ class AgentKitAPI:
     def create_memory_store(
         self, managed_memory_store: ManagedMemoryStore, managed_memory_store_id: str
     ) -> ManagedMemoryStore:
-        """Creates a managed memory store in the caller's workspace.
+        """Creates a memory store in the caller's workspace.
 
         :param managed_memory_store: :class:`ManagedMemoryStore`
-          The managed memory store to create.
+          The memory store to create.
         :param managed_memory_store_id: str
-          Caller-provided, workspace-unique managed memory store ID. It must be 3-56 characters, begin with a
+          Caller-provided, workspace-unique memory store ID. It must be 3-56 characters, begin with a
           lowercase letter, contain only lowercase letters, digits, and hyphens, and end with a letter or
           digit.
 
@@ -1075,8 +1074,8 @@ class AgentKitAPI:
         return SessionStore.from_dict(res)
 
     def delete_memory(self, name: str):
-        """Deletes a managed memory entry by resource name. Returns ``NOT_FOUND`` when the entry does not exist
-        in the caller's workspace.
+        """Deletes a memory entry by resource name. Returns ``NOT_FOUND`` when the entry does not exist in the
+        caller's workspace.
 
         :param name: str
           Resource name in the form
@@ -1096,8 +1095,8 @@ class AgentKitAPI:
         self._api.do("DELETE", f"/api/2.0/agents/{name}", headers=headers)
 
     def delete_memory_store(self, name: str):
-        """Deletes a managed memory store by resource name. Returns ``NOT_FOUND`` when the store does not exist
-        in the caller's workspace.
+        """Deletes a memory store by resource name. Returns ``NOT_FOUND`` when the store does not exist in the
+        caller's workspace.
 
         :param name: str
           Resource name in the form ``memory-stores/{managed_memory_store_id}``.
@@ -1172,7 +1171,7 @@ class AgentKitAPI:
         :param session_id: str
           Identifier of the session whose transcript is distilled into memories.
         :param memory_store: str
-          Managed memory store the extracted entries are written to, in the form
+          Memory store the extracted entries are written to, in the form
           ``memory-stores/{managed_memory_store_id}``.
         :param dry_run: bool (optional)
           When true, extract and return the entries without writing them to the memory store. Defaults to
@@ -1257,8 +1256,8 @@ class AgentKitAPI:
         return ForkSessionResponse.from_dict(res)
 
     def get_memory(self, name: str, *, view: Optional[ManagedMemoryEntryView] = None) -> ManagedMemoryEntry:
-        """Retrieves a managed memory entry by resource name. ``view`` controls whether the response includes
-        memory content. Returns ``NOT_FOUND`` when the entry does not exist in the caller's workspace.
+        """Retrieves a memory entry by resource name. ``view`` controls whether the response includes memory
+        content. Returns ``NOT_FOUND`` when the entry does not exist in the caller's workspace.
 
         :param name: str
           Resource name in the form
@@ -1284,8 +1283,8 @@ class AgentKitAPI:
         return ManagedMemoryEntry.from_dict(res)
 
     def get_memory_store(self, name: str) -> ManagedMemoryStore:
-        """Retrieves a managed memory store by resource name. Returns ``NOT_FOUND`` when the store does not exist
-        in the caller's workspace.
+        """Retrieves a memory store by resource name. Returns ``NOT_FOUND`` when the store does not exist in the
+        caller's workspace.
 
         :param name: str
           Resource name in the form ``memory-stores/{managed_memory_store_id}``.
@@ -1355,13 +1354,12 @@ class AgentKitAPI:
         session_id: Optional[str] = None,
         view: Optional[ManagedMemoryEntryView] = None,
     ) -> Iterator[ManagedMemoryEntry]:
-        """Lists managed memory entries for one actor. An exact ``path`` filters entries across sessions,
-        ignoring session metadata. Otherwise, ``session_id`` and ``path_prefix`` restrict the actor partition.
-        ``view`` controls whether each returned entry includes memory content.
+        """Lists memory entries for one actor. An exact ``path`` filters entries across sessions, ignoring
+        session metadata. Otherwise, ``session_id`` and ``path_prefix`` restrict the actor partition. ``view``
+        controls whether each returned entry includes memory content.
 
         :param parent: str
-          Managed memory store whose entries are listed, in the form
-          ``memory-stores/{managed_memory_store_id}``.
+          Memory store whose entries are listed, in the form ``memory-stores/{managed_memory_store_id}``.
         :param actor_id: str
           Customer-provided identifier for the actor whose entries are listed.
         :param page_size: int (optional)
@@ -1413,7 +1411,7 @@ class AgentKitAPI:
     def list_memory_stores(
         self, *, page_size: Optional[int] = None, page_token: Optional[str] = None
     ) -> Iterator[ManagedMemoryStore]:
-        """Lists managed memory stores in the caller's workspace.
+        """Lists memory stores in the caller's workspace.
 
         :param page_size: int (optional)
           Maximum number of stores to return. The service may return fewer stores than requested. Defaults to
@@ -1621,12 +1619,11 @@ class AgentKitAPI:
         session_id: Optional[str] = None,
         view: Optional[ManagedMemoryEntryView] = None,
     ) -> Iterator[ManagedMemoryEntrySearchResult]:
-        """Searches managed memory entries by text query for one actor. Returns matching entries and scores
-        ranked by relevance; ``view`` controls whether each returned entry includes content.
+        """Searches memory entries by text query for one actor. Returns matching entries and scores ranked by
+        relevance; ``view`` controls whether each returned entry includes content.
 
         :param parent: str
-          Managed memory store whose entries are searched, in the form
-          ``memory-stores/{managed_memory_store_id}``.
+          Memory store whose entries are searched, in the form ``memory-stores/{managed_memory_store_id}``.
         :param actor_id: str
           Customer-provided identifier for the actor whose entries are searched.
         :param query: str
@@ -1689,13 +1686,13 @@ class AgentKitAPI:
     def update_memory(
         self, name: str, managed_memory_entry: ManagedMemoryEntry, update_mask: FieldMask
     ) -> ManagedMemoryEntry:
-        """Updates selected mutable fields on a managed memory entry. Identity fields are immutable.
+        """Updates selected mutable fields on a memory entry. Identity fields are immutable.
 
         :param name: str
           Resource name in the form
           ``memory-stores/{managed_memory_store_id}/entries/{managed_memory_entry_id}``.
         :param managed_memory_entry: :class:`ManagedMemoryEntry`
-          The managed memory entry to update.
+          The memory entry to update.
         :param update_mask: FieldMask
           Fields to update. Only ``content`` and ``description`` may be updated.
 
@@ -1721,12 +1718,12 @@ class AgentKitAPI:
     def update_memory_store(
         self, name: str, managed_memory_store: ManagedMemoryStore, update_mask: FieldMask
     ) -> ManagedMemoryStore:
-        """Updates a managed memory store's description and metadata.
+        """Updates a memory store's description and metadata.
 
         :param name: str
           Resource name in the form ``memory-stores/{managed_memory_store_id}``.
         :param managed_memory_store: :class:`ManagedMemoryStore`
-          The managed memory store to update. ``name`` is taken from the URL.
+          The memory store to update. ``name`` is taken from the URL.
         :param update_mask: FieldMask
           Only ``description`` and ``metadata`` may be updated. Selecting ``metadata`` replaces the entire
           map.

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/databricks/sdk/service/catalog.py

@@ -1765,6 +1765,10 @@ class ConnectionInfo:
     read_only: Optional[bool] = None
     """If the connection is read only."""
 
+    secrets: Optional[Dict[str, str]] = None
+    """A map of option names to UC Secret references. Keys are connection option names (same as in
+    OptionsKVPairs) and values are UC Secret fully qualified names."""
+
     securable_type: Optional[SecurableType] = None
 
     updated_at: Optional[int] = None
@@ -1809,6 +1813,8 @@ class ConnectionInfo:
             body["provisioning_info"] = self.provisioning_info.as_dict()
         if self.read_only is not None:
             body["read_only"] = self.read_only
+        if self.secrets:
+            body["secrets"] = self.secrets
         if self.securable_type is not None:
             body["securable_type"] = self.securable_type.value
         if self.updated_at is not None:
@@ -1852,6 +1858,8 @@ class ConnectionInfo:
             body["provisioning_info"] = self.provisioning_info
         if self.read_only is not None:
             body["read_only"] = self.read_only
+        if self.secrets:
+            body["secrets"] = self.secrets
         if self.securable_type is not None:
             body["securable_type"] = self.securable_type
         if self.updated_at is not None:
@@ -1881,6 +1889,7 @@ class ConnectionInfo:
             properties=d.get("properties", None),
             provisioning_info=_from_dict(d, "provisioning_info", ProvisioningInfo),
             read_only=d.get("read_only", None),
+            secrets=d.get("secrets", None),
             securable_type=_enum(d, "securable_type", SecurableType),
             updated_at=_int64(d, "updated_at"),
             updated_by=d.get("updated_by", None),
@@ -5757,6 +5766,40 @@ class GetWorkspaceBindingsResponse:
         )
 
 
+@dataclass
+class GovernedTagReference:
+    """A governed tag referenced by a policy condition."""
+
+    tag_key: str
+    """The governed tag key."""
+
+    tag_value: Optional[str] = None
+    """The governed tag value, when the policy references a specific value."""
+
+    def as_dict(self) -> dict:
+        """Serializes the GovernedTagReference into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.tag_key is not None:
+            body["tag_key"] = self.tag_key
+        if self.tag_value is not None:
+            body["tag_value"] = self.tag_value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the GovernedTagReference into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.tag_key is not None:
+            body["tag_key"] = self.tag_key
+        if self.tag_value is not None:
+            body["tag_value"] = self.tag_value
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> GovernedTagReference:
+        """Deserializes the GovernedTagReference from a dictionary."""
+        return cls(tag_key=d.get("tag_key", None), tag_value=d.get("tag_value", None))
+
+
 @dataclass
 class GrantOptions:
     privileges: List[str]
@@ -10546,6 +10589,10 @@ class PolicyInfo:
     except_principals: Optional[List[str]] = None
     """Optional list of user or group names that should be excluded from the policy."""
 
+    governed_tags: Optional[List[GovernedTagReference]] = None
+    """Governed tag keys and values referenced by this policy's conditions and function arguments.
+    Output only."""
+
     grant: Optional[GrantOptions] = None
     """Options for grant policies. Valid only if ``policy_type`` is ``POLICY_TYPE_GRANT``. Required on
     create and optional on update. When specified on update, the new options will replace the
@@ -10601,6 +10648,8 @@ class PolicyInfo:
             body["except_principals"] = [v for v in self.except_principals]
         if self.for_securable_type is not None:
             body["for_securable_type"] = self.for_securable_type.value
+        if self.governed_tags:
+            body["governed_tags"] = [v.as_dict() for v in self.governed_tags]
         if self.grant:
             body["grant"] = self.grant.as_dict()
         if self.id is not None:
@@ -10644,6 +10693,8 @@ class PolicyInfo:
             body["except_principals"] = self.except_principals
         if self.for_securable_type is not None:
             body["for_securable_type"] = self.for_securable_type
+        if self.governed_tags:
+            body["governed_tags"] = self.governed_tags
         if self.grant:
             body["grant"] = self.grant
         if self.id is not None:
@@ -10681,6 +10732,7 @@ class PolicyInfo:
             deny=_from_dict(d, "deny", DenyOptions),
             except_principals=d.get("except_principals", None),
             for_securable_type=_enum(d, "for_securable_type", SecurableType),
+            governed_tags=_repeated_dict(d, "governed_tags", GovernedTagReference),
             grant=_from_dict(d, "grant", GrantOptions),
             id=d.get("id", None),
             match_columns=_repeated_dict(d, "match_columns", MatchColumn),
@@ -15574,6 +15626,7 @@ class ConnectionsAPI:
         parent: Optional[str] = None,
         properties: Optional[Dict[str, str]] = None,
         read_only: Optional[bool] = None,
+        secrets: Optional[Dict[str, str]] = None,
     ) -> ConnectionInfo:
         """Creates a new connection
 
@@ -15597,6 +15650,9 @@ class ConnectionsAPI:
           A map of key-value properties attached to the securable.
         :param read_only: bool (optional)
           If the connection is read only.
+        :param secrets: Dict[str,str] (optional)
+          A map of option names to UC Secret references. Keys are connection option names (same as in
+          OptionsKVPairs) and values are UC Secret fully qualified names.
 
         :returns: :class:`ConnectionInfo`
         """
@@ -15618,6 +15674,8 @@ class ConnectionsAPI:
             body["properties"] = properties
         if read_only is not None:
             body["read_only"] = read_only
+        if secrets is not None:
+            body["secrets"] = secrets
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",
@@ -15732,6 +15790,7 @@ class ConnectionsAPI:
         environment_settings: Optional[EnvironmentSettings] = None,
         new_name: Optional[str] = None,
         owner: Optional[str] = None,
+        secrets: Optional[Dict[str, str]] = None,
     ) -> ConnectionInfo:
         """Updates the connection that matches the supplied name.
 
@@ -15745,6 +15804,9 @@ class ConnectionsAPI:
           New name for the connection.
         :param owner: str (optional)
           Username of current owner of the connection.
+        :param secrets: Dict[str,str] (optional)
+          A map of option names to UC Secret references. Keys are connection option names (same as in
+          OptionsKVPairs) and values are UC Secret fully qualified names.
 
         :returns: :class:`ConnectionInfo`
         """
@@ -15758,6 +15820,8 @@ class ConnectionsAPI:
             body["options"] = options
         if owner is not None:
             body["owner"] = owner
+        if secrets is not None:
+            body["secrets"] = secrets
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/databricks/sdk/service/compute.py

@@ -675,11 +675,7 @@ class ClusterAttributes:
 
     driver_node_type_id: Optional[str] = None
     """The node type of the Spark driver. Note that this field is optional; if unset, the driver node
-    type will be set as the same value as ``node_type_id`` defined above.
-    
-    This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-    driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id
-    and node_type_id take precedence."""
+    type will be set as the same value as ``node_type_id`` defined above."""
 
     enable_elastic_disk: Optional[bool] = None
     """Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk
@@ -1098,11 +1094,7 @@ class ClusterDetails:
 
     driver_node_type_id: Optional[str] = None
     """The node type of the Spark driver. Note that this field is optional; if unset, the driver node
-    type will be set as the same value as ``node_type_id`` defined above.
-    
-    This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-    driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id
-    and node_type_id take precedence."""
+    type will be set as the same value as ``node_type_id`` defined above."""
 
     enable_elastic_disk: Optional[bool] = None
     """Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk
@@ -2187,11 +2179,7 @@ class ClusterSpec:
 
     driver_node_type_id: Optional[str] = None
     """The node type of the Spark driver. Note that this field is optional; if unset, the driver node
-    type will be set as the same value as ``node_type_id`` defined above.
-    
-    This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-    driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id
-    and node_type_id take precedence."""
+    type will be set as the same value as ``node_type_id`` defined above."""
 
     enable_elastic_disk: Optional[bool] = None
     """Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk
@@ -3360,11 +3348,7 @@ class EnforcePolicyComplianceForClusterResponseClusterSettings:
 
     driver_node_type_id: Optional[str] = None
     """The node type of the Spark driver. Note that this field is optional; if unset, the driver node
-    type will be set as the same value as ``node_type_id`` defined above.
-    
-    This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-    driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id
-    and node_type_id take precedence."""
+    type will be set as the same value as ``node_type_id`` defined above."""
 
     enable_elastic_disk: Optional[bool] = None
     """Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk
@@ -8146,11 +8130,7 @@ class UpdateClusterResource:
 
     driver_node_type_id: Optional[str] = None
     """The node type of the Spark driver. Note that this field is optional; if unset, the driver node
-    type will be set as the same value as ``node_type_id`` defined above.
-    
-    This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-    driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id
-    and node_type_id take precedence."""
+    type will be set as the same value as ``node_type_id`` defined above."""
 
     enable_elastic_disk: Optional[bool] = None
     """Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk
@@ -9149,10 +9129,6 @@ class ClustersAPI:
         :param driver_node_type_id: str (optional)
           The node type of the Spark driver. Note that this field is optional; if unset, the driver node type
           will be set as the same value as ``node_type_id`` defined above.
-
-          This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-          driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id and
-          node_type_id take precedence.
         :param enable_elastic_disk: bool (optional)
           Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk space
           when its Spark workers are running low on disk space.
@@ -9542,10 +9518,6 @@ class ClustersAPI:
         :param driver_node_type_id: str (optional)
           The node type of the Spark driver. Note that this field is optional; if unset, the driver node type
           will be set as the same value as ``node_type_id`` defined above.
-
-          This field, along with node_type_id, should not be set if virtual_cluster_size is set. If both
-          driver_node_type_id, node_type_id, and virtual_cluster_size are specified, driver_node_type_id and
-          node_type_id take precedence.
         :param enable_elastic_disk: bool (optional)
           Autoscaling Local Storage: when enabled, this cluster will dynamically acquire additional disk space
           when its Spark workers are running low on disk space.

{databricks_sdk-0.153.0 → databricks_sdk-0.154.0}/databricks/sdk/service/disasterrecovery.py

@@ -58,6 +58,13 @@ class FailoverGroup:
     etag: Optional[str] = None
     """Opaque version string for optimistic locking. Server-generated and returned in responses."""
 
+    has_replication_point_blocking_errors: Optional[bool] = None
+    """True if errors currently block the replication point."""
+
+    initial_replication_progress: Optional[InitialReplicationProgress] = None
+    """Initial replication progress of all in-scope assets. Set only when the FailoverGroup is in
+    INITIAL_REPLICATION state."""
+
     name: Optional[str] = None
     """Fully qualified resource name in the format
     accounts/{account_id}/failover-groups/{failover_group_id}."""
@@ -83,8 +90,12 @@ class FailoverGroup:
             body["effective_primary_region"] = self.effective_primary_region
         if self.etag is not None:
             body["etag"] = self.etag
+        if self.has_replication_point_blocking_errors is not None:
+            body["has_replication_point_blocking_errors"] = self.has_replication_point_blocking_errors
         if self.initial_primary_region is not None:
             body["initial_primary_region"] = self.initial_primary_region
+        if self.initial_replication_progress:
+            body["initial_replication_progress"] = self.initial_replication_progress.as_dict()
         if self.name is not None:
             body["name"] = self.name
         if self.regions:
@@ -110,8 +121,12 @@ class FailoverGroup:
             body["effective_primary_region"] = self.effective_primary_region
         if self.etag is not None:
             body["etag"] = self.etag
+        if self.has_replication_point_blocking_errors is not None:
+            body["has_replication_point_blocking_errors"] = self.has_replication_point_blocking_errors
         if self.initial_primary_region is not None:
             body["initial_primary_region"] = self.initial_primary_region
+        if self.initial_replication_progress:
+            body["initial_replication_progress"] = self.initial_replication_progress
         if self.name is not None:
             body["name"] = self.name
         if self.regions:
@@ -135,7 +150,9 @@ class FailoverGroup:
             create_time=_timestamp(d, "create_time"),
             effective_primary_region=d.get("effective_primary_region", None),
             etag=d.get("etag", None),
+            has_replication_point_blocking_errors=d.get("has_replication_point_blocking_errors", None),
             initial_primary_region=d.get("initial_primary_region", None),
+            initial_replication_progress=_from_dict(d, "initial_replication_progress", InitialReplicationProgress),
             name=d.get("name", None),
             regions=d.get("regions", None),
             replication_point=_timestamp(d, "replication_point"),
@@ -159,6 +176,34 @@ class FailoverGroupState(Enum):
     INITIAL_REPLICATION = "INITIAL_REPLICATION"
 
 
+@dataclass
+class InitialReplicationProgress:
+    """Progress of the initial replication of all in-scope assets (Unity Catalog and workspace assets)
+    of a FailoverGroup. Populated only while the FailoverGroup is in INITIAL_REPLICATION."""
+
+    completed: Optional[int] = None
+    """Percent complete, 0-100."""
+
+    def as_dict(self) -> dict:
+        """Serializes the InitialReplicationProgress into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.completed is not None:
+            body["completed"] = self.completed
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the InitialReplicationProgress into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.completed is not None:
+            body["completed"] = self.completed
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> InitialReplicationProgress:
+        """Deserializes the InitialReplicationProgress from a dictionary."""
+        return cls(completed=d.get("completed", None))
+
+
 @dataclass
 class ListFailoverGroupsResponse:
     """Response for listing failover groups."""