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

databricks/sdk/service/database.py

@@ -3,11 +3,14 @@
 from __future__ import annotations
 
 import logging
+import random
+import time
 from dataclasses import dataclass
+from datetime import timedelta
 from enum import Enum
-from typing import Any, Dict, Iterator, List, Optional
+from typing import Any, Callable, Dict, Iterator, List, Optional
 
-from ._internal import _enum, _from_dict, _repeated_dict
+from ._internal import Wait, _enum, _from_dict, _repeated_dict
 
 _LOG = logging.getLogger("databricks.sdk")
 
@@ -787,7 +790,7 @@ class RequestedResource:
 
 @dataclass
 class SyncedDatabaseTable:
-    """Next field marker: 12"""
+    """Next field marker: 14"""
 
     name: str
     """Full three-part (catalog, schema, table) name of the table."""
@@ -802,6 +805,14 @@ class SyncedDatabaseTable:
     database instance name MUST match that of the registered catalog (or the request will be
     rejected)."""
 
+    effective_database_instance_name: Optional[str] = None
+    """The name of the database instance that this table is registered to. This field is always
+    returned, and for tables inside database catalogs is inferred database instance associated with
+    the catalog."""
+
+    effective_logical_database_name: Optional[str] = None
+    """The name of the logical database that this table is registered to."""
+
     logical_database_name: Optional[str] = None
     """Target Postgres database object (logical database) name for this table.
     
@@ -828,6 +839,10 @@ class SyncedDatabaseTable:
             body["data_synchronization_status"] = self.data_synchronization_status.as_dict()
         if self.database_instance_name is not None:
             body["database_instance_name"] = self.database_instance_name
+        if self.effective_database_instance_name is not None:
+            body["effective_database_instance_name"] = self.effective_database_instance_name
+        if self.effective_logical_database_name is not None:
+            body["effective_logical_database_name"] = self.effective_logical_database_name
         if self.logical_database_name is not None:
             body["logical_database_name"] = self.logical_database_name
         if self.name is not None:
@@ -845,6 +860,10 @@ class SyncedDatabaseTable:
             body["data_synchronization_status"] = self.data_synchronization_status
         if self.database_instance_name is not None:
             body["database_instance_name"] = self.database_instance_name
+        if self.effective_database_instance_name is not None:
+            body["effective_database_instance_name"] = self.effective_database_instance_name
+        if self.effective_logical_database_name is not None:
+            body["effective_logical_database_name"] = self.effective_logical_database_name
         if self.logical_database_name is not None:
             body["logical_database_name"] = self.logical_database_name
         if self.name is not None:
@@ -861,6 +880,8 @@ class SyncedDatabaseTable:
         return cls(
             data_synchronization_status=_from_dict(d, "data_synchronization_status", SyncedTableStatus),
             database_instance_name=d.get("database_instance_name", None),
+            effective_database_instance_name=d.get("effective_database_instance_name", None),
+            effective_logical_database_name=d.get("effective_logical_database_name", None),
             logical_database_name=d.get("logical_database_name", None),
             name=d.get("name", None),
             spec=_from_dict(d, "spec", SyncedTableSpec),
@@ -1358,6 +1379,31 @@ class DatabaseAPI:
     def __init__(self, api_client):
         self._api = api_client
 
+    def wait_get_database_instance_database_available(
+        self, name: str, timeout=timedelta(minutes=20), callback: Optional[Callable[[DatabaseInstance], None]] = None
+    ) -> DatabaseInstance:
+        deadline = time.time() + timeout.total_seconds()
+        target_states = (DatabaseInstanceState.AVAILABLE,)
+        status_message = "polling..."
+        attempt = 1
+        while time.time() < deadline:
+            poll = self.get_database_instance(name=name)
+            status = poll.state
+            status_message = f"current status: {status}"
+            if status in target_states:
+                return poll
+            if callback:
+                callback(poll)
+            prefix = f"name={name}"
+            sleep = attempt
+            if sleep > 10:
+                # sleep 10s max per attempt
+                sleep = 10
+            _LOG.debug(f"{prefix}: ({status}) {status_message} (sleeping ~{sleep}s)")
+            time.sleep(sleep + random.random())
+            attempt += 1
+        raise TimeoutError(f"timed out after {timeout}: {status_message}")
+
     def create_database_catalog(self, catalog: DatabaseCatalog) -> DatabaseCatalog:
         """Create a Database Catalog.
 
@@ -1374,13 +1420,15 @@ class DatabaseAPI:
         res = self._api.do("POST", "/api/2.0/database/catalogs", body=body, headers=headers)
         return DatabaseCatalog.from_dict(res)
 
-    def create_database_instance(self, database_instance: DatabaseInstance) -> DatabaseInstance:
+    def create_database_instance(self, database_instance: DatabaseInstance) -> Wait[DatabaseInstance]:
         """Create a Database Instance.
 
         :param database_instance: :class:`DatabaseInstance`
           Instance to create.
 
-        :returns: :class:`DatabaseInstance`
+        :returns:
+          Long-running operation waiter for :class:`DatabaseInstance`.
+          See :method:wait_get_database_instance_database_available for more details.
         """
         body = database_instance.as_dict()
         headers = {
@@ -1388,8 +1436,17 @@ class DatabaseAPI:
             "Content-Type": "application/json",
         }
 
-        res = self._api.do("POST", "/api/2.0/database/instances", body=body, headers=headers)
-        return DatabaseInstance.from_dict(res)
+        op_response = self._api.do("POST", "/api/2.0/database/instances", body=body, headers=headers)
+        return Wait(
+            self.wait_get_database_instance_database_available,
+            response=DatabaseInstance.from_dict(op_response),
+            name=op_response["name"],
+        )
+
+    def create_database_instance_and_wait(
+        self, database_instance: DatabaseInstance, timeout=timedelta(minutes=20)
+    ) -> DatabaseInstance:
+        return self.create_database_instance(database_instance=database_instance).result(timeout=timeout)
 
     def create_database_instance_role(
         self, instance_name: str, database_instance_role: DatabaseInstanceRole

databricks/sdk/service/jobs.py

@@ -7223,6 +7223,73 @@ class SubscriptionSubscriber:
         return cls(destination_id=d.get("destination_id", None), user_name=d.get("user_name", None))
 
 
+@dataclass
+class TableState:
+    has_seen_updates: Optional[bool] = None
+    """Whether or not the table has seen updates since either the creation of the trigger or the last
+    successful evaluation of the trigger"""
+
+    table_name: Optional[str] = None
+    """Full table name of the table to monitor, e.g. `mycatalog.myschema.mytable`"""
+
+    def as_dict(self) -> dict:
+        """Serializes the TableState into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.has_seen_updates is not None:
+            body["has_seen_updates"] = self.has_seen_updates
+        if self.table_name is not None:
+            body["table_name"] = self.table_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TableState into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.has_seen_updates is not None:
+            body["has_seen_updates"] = self.has_seen_updates
+        if self.table_name is not None:
+            body["table_name"] = self.table_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TableState:
+        """Deserializes the TableState from a dictionary."""
+        return cls(has_seen_updates=d.get("has_seen_updates", None), table_name=d.get("table_name", None))
+
+
+@dataclass
+class TableTriggerState:
+    last_seen_table_states: Optional[List[TableState]] = None
+
+    using_scalable_monitoring: Optional[bool] = None
+    """Indicates whether the trigger is using scalable monitoring."""
+
+    def as_dict(self) -> dict:
+        """Serializes the TableTriggerState into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.last_seen_table_states:
+            body["last_seen_table_states"] = [v.as_dict() for v in self.last_seen_table_states]
+        if self.using_scalable_monitoring is not None:
+            body["using_scalable_monitoring"] = self.using_scalable_monitoring
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TableTriggerState into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.last_seen_table_states:
+            body["last_seen_table_states"] = self.last_seen_table_states
+        if self.using_scalable_monitoring is not None:
+            body["using_scalable_monitoring"] = self.using_scalable_monitoring
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TableTriggerState:
+        """Deserializes the TableTriggerState from a dictionary."""
+        return cls(
+            last_seen_table_states=_repeated_dict(d, "last_seen_table_states", TableState),
+            using_scalable_monitoring=d.get("using_scalable_monitoring", None),
+        )
+
+
 @dataclass
 class TableUpdateTriggerConfiguration:
     condition: Optional[Condition] = None
@@ -7993,11 +8060,15 @@ class TriggerSettings:
 class TriggerStateProto:
     file_arrival: Optional[FileArrivalTriggerState] = None
 
+    table: Optional[TableTriggerState] = None
+
     def as_dict(self) -> dict:
         """Serializes the TriggerStateProto into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.file_arrival:
             body["file_arrival"] = self.file_arrival.as_dict()
+        if self.table:
+            body["table"] = self.table.as_dict()
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -8005,12 +8076,17 @@ class TriggerStateProto:
         body = {}
         if self.file_arrival:
             body["file_arrival"] = self.file_arrival
+        if self.table:
+            body["table"] = self.table
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> TriggerStateProto:
         """Deserializes the TriggerStateProto from a dictionary."""
-        return cls(file_arrival=_from_dict(d, "file_arrival", FileArrivalTriggerState))
+        return cls(
+            file_arrival=_from_dict(d, "file_arrival", FileArrivalTriggerState),
+            table=_from_dict(d, "table", TableTriggerState),
+        )
 
 
 class TriggerType(Enum):

databricks/sdk/service/serving.py

@@ -205,6 +205,7 @@ class AiGatewayGuardrailPiiBehavior:
 class AiGatewayGuardrailPiiBehaviorBehavior(Enum):
 
     BLOCK = "BLOCK"
+    MASK = "MASK"
     NONE = "NONE"
 
 
@@ -989,10 +990,13 @@ class DatabricksModelServingConfig:
 @dataclass
 class DataframeSplitInput:
     columns: Optional[List[Any]] = None
+    """Columns array for the dataframe"""
 
     data: Optional[List[Any]] = None
+    """Data array for the dataframe"""
 
     index: Optional[List[int]] = None
+    """Index array for the dataframe"""
 
     def as_dict(self) -> dict:
         """Serializes the DataframeSplitInput into a dictionary suitable for use as a JSON request body."""
@@ -1040,9 +1044,46 @@ class DeleteResponse:
         return cls()
 
 
+@dataclass
+class EmailNotifications:
+    on_update_failure: Optional[List[str]] = None
+    """A list of email addresses to be notified when an endpoint fails to update its configuration or
+    state."""
+
+    on_update_success: Optional[List[str]] = None
+    """A list of email addresses to be notified when an endpoint successfully updates its configuration
+    or state."""
+
+    def as_dict(self) -> dict:
+        """Serializes the EmailNotifications into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.on_update_failure:
+            body["on_update_failure"] = [v for v in self.on_update_failure]
+        if self.on_update_success:
+            body["on_update_success"] = [v for v in self.on_update_success]
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the EmailNotifications into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.on_update_failure:
+            body["on_update_failure"] = self.on_update_failure
+        if self.on_update_success:
+            body["on_update_success"] = self.on_update_success
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> EmailNotifications:
+        """Deserializes the EmailNotifications from a dictionary."""
+        return cls(
+            on_update_failure=d.get("on_update_failure", None), on_update_success=d.get("on_update_success", None)
+        )
+
+
 @dataclass
 class EmbeddingsV1ResponseEmbeddingElement:
     embedding: Optional[List[float]] = None
+    """The embedding vector"""
 
     index: Optional[int] = None
     """The index of the embedding in the response."""
@@ -3261,11 +3302,11 @@ class ServedModelState:
 
 class ServedModelStateDeployment(Enum):
 
-    ABORTED = "DEPLOYMENT_ABORTED"
-    CREATING = "DEPLOYMENT_CREATING"
-    FAILED = "DEPLOYMENT_FAILED"
-    READY = "DEPLOYMENT_READY"
-    RECOVERING = "DEPLOYMENT_RECOVERING"
+    DEPLOYMENT_ABORTED = "DEPLOYMENT_ABORTED"
+    DEPLOYMENT_CREATING = "DEPLOYMENT_CREATING"
+    DEPLOYMENT_FAILED = "DEPLOYMENT_FAILED"
+    DEPLOYMENT_READY = "DEPLOYMENT_READY"
+    DEPLOYMENT_RECOVERING = "DEPLOYMENT_RECOVERING"
 
 
 @dataclass
@@ -3544,6 +3585,9 @@ class ServingEndpointDetailed:
     description: Optional[str] = None
     """Description of the serving model"""
 
+    email_notifications: Optional[EmailNotifications] = None
+    """Email notification settings."""
+
     endpoint_url: Optional[str] = None
     """Endpoint invocation url if route optimization is enabled for endpoint"""
 
@@ -3592,6 +3636,8 @@ class ServingEndpointDetailed:
             body["data_plane_info"] = self.data_plane_info.as_dict()
         if self.description is not None:
             body["description"] = self.description
+        if self.email_notifications:
+            body["email_notifications"] = self.email_notifications.as_dict()
         if self.endpoint_url is not None:
             body["endpoint_url"] = self.endpoint_url
         if self.id is not None:
@@ -3631,6 +3677,8 @@ class ServingEndpointDetailed:
             body["data_plane_info"] = self.data_plane_info
         if self.description is not None:
             body["description"] = self.description
+        if self.email_notifications:
+            body["email_notifications"] = self.email_notifications
         if self.endpoint_url is not None:
             body["endpoint_url"] = self.endpoint_url
         if self.id is not None:
@@ -3664,6 +3712,7 @@ class ServingEndpointDetailed:
             creator=d.get("creator", None),
             data_plane_info=_from_dict(d, "data_plane_info", ModelDataPlaneInfo),
             description=d.get("description", None),
+            email_notifications=_from_dict(d, "email_notifications", EmailNotifications),
             endpoint_url=d.get("endpoint_url", None),
             id=d.get("id", None),
             last_updated_timestamp=d.get("last_updated_timestamp", None),
@@ -3977,6 +4026,7 @@ class ServingEndpointsAPI:
         budget_policy_id: Optional[str] = None,
         config: Optional[EndpointCoreConfigInput] = None,
         description: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         rate_limits: Optional[List[RateLimit]] = None,
         route_optimized: Optional[bool] = None,
         tags: Optional[List[EndpointTag]] = None,
@@ -3995,6 +4045,8 @@ class ServingEndpointsAPI:
         :param config: :class:`EndpointCoreConfigInput` (optional)
           The core config of the serving endpoint.
         :param description: str (optional)
+        :param email_notifications: :class:`EmailNotifications` (optional)
+          Email notification settings.
         :param rate_limits: List[:class:`RateLimit`] (optional)
           Rate limits to be applied to the serving endpoint. NOTE: this field is deprecated, please use AI
           Gateway to manage rate limits.
@@ -4016,6 +4068,8 @@ class ServingEndpointsAPI:
             body["config"] = config.as_dict()
         if description is not None:
             body["description"] = description
+        if email_notifications is not None:
+            body["email_notifications"] = email_notifications.as_dict()
         if name is not None:
             body["name"] = name
         if rate_limits is not None:
@@ -4044,6 +4098,7 @@ class ServingEndpointsAPI:
         budget_policy_id: Optional[str] = None,
         config: Optional[EndpointCoreConfigInput] = None,
         description: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         rate_limits: Optional[List[RateLimit]] = None,
         route_optimized: Optional[bool] = None,
         tags: Optional[List[EndpointTag]] = None,
@@ -4054,6 +4109,7 @@ class ServingEndpointsAPI:
             budget_policy_id=budget_policy_id,
             config=config,
             description=description,
+            email_notifications=email_notifications,
             name=name,
             rate_limits=rate_limits,
             route_optimized=route_optimized,
@@ -4067,6 +4123,7 @@ class ServingEndpointsAPI:
         *,
         ai_gateway: Optional[AiGatewayConfig] = None,
         budget_policy_id: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         tags: Optional[List[EndpointTag]] = None,
     ) -> Wait[ServingEndpointDetailed]:
         """Create a new PT serving endpoint.
@@ -4080,6 +4137,8 @@ class ServingEndpointsAPI:
           The AI Gateway configuration for the serving endpoint.
         :param budget_policy_id: str (optional)
           The budget policy associated with the endpoint.
+        :param email_notifications: :class:`EmailNotifications` (optional)
+          Email notification settings.
         :param tags: List[:class:`EndpointTag`] (optional)
           Tags to be attached to the serving endpoint and automatically propagated to billing logs.
 
@@ -4094,6 +4153,8 @@ class ServingEndpointsAPI:
             body["budget_policy_id"] = budget_policy_id
         if config is not None:
             body["config"] = config.as_dict()
+        if email_notifications is not None:
+            body["email_notifications"] = email_notifications.as_dict()
         if name is not None:
             body["name"] = name
         if tags is not None:
@@ -4117,11 +4178,17 @@ class ServingEndpointsAPI:
         *,
         ai_gateway: Optional[AiGatewayConfig] = None,
         budget_policy_id: Optional[str] = None,
+        email_notifications: Optional[EmailNotifications] = None,
         tags: Optional[List[EndpointTag]] = None,
         timeout=timedelta(minutes=20),
     ) -> ServingEndpointDetailed:
         return self.create_provisioned_throughput_endpoint(
-            ai_gateway=ai_gateway, budget_policy_id=budget_policy_id, config=config, name=name, tags=tags
+            ai_gateway=ai_gateway,
+            budget_policy_id=budget_policy_id,
+            config=config,
+            email_notifications=email_notifications,
+            name=name,
+            tags=tags,
         ).result(timeout=timeout)
 
     def delete(self, name: str):
@@ -4421,10 +4488,10 @@ class ServingEndpointsAPI:
         stream: Optional[bool] = None,
         temperature: Optional[float] = None,
     ) -> QueryEndpointResponse:
-        """Query a serving endpoint.
+        """Query a serving endpoint
 
         :param name: str
-          The name of the serving endpoint. This field is required.
+          The name of the serving endpoint. This field is required and is provided via the path parameter.
         :param dataframe_records: List[Any] (optional)
           Pandas Dataframe input in the records orientation.
         :param dataframe_split: :class:`DataframeSplitInput` (optional)
@@ -4445,8 +4512,8 @@ class ServingEndpointsAPI:
           The max tokens field used ONLY for __completions__ and __chat external & foundation model__ serving
           endpoints. This is an integer and should only be used with other chat/completions query fields.
         :param messages: List[:class:`ChatMessage`] (optional)
-          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is a
-          map of strings and should only be used with other chat query fields.
+          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is an
+          array of ChatMessage objects and should only be used with other chat query fields.
         :param n: int (optional)
           The n (number of candidates) field used ONLY for __completions__ and __chat external & foundation
           model__ serving endpoints. This is an integer between 1 and 5 with a default of 1 and should only be
@@ -4724,10 +4791,10 @@ class ServingEndpointsDataPlaneAPI:
         stream: Optional[bool] = None,
         temperature: Optional[float] = None,
     ) -> QueryEndpointResponse:
-        """Query a serving endpoint.
+        """Query a serving endpoint
 
         :param name: str
-          The name of the serving endpoint. This field is required.
+          The name of the serving endpoint. This field is required and is provided via the path parameter.
         :param dataframe_records: List[Any] (optional)
           Pandas Dataframe input in the records orientation.
         :param dataframe_split: :class:`DataframeSplitInput` (optional)
@@ -4748,8 +4815,8 @@ class ServingEndpointsDataPlaneAPI:
           The max tokens field used ONLY for __completions__ and __chat external & foundation model__ serving
           endpoints. This is an integer and should only be used with other chat/completions query fields.
         :param messages: List[:class:`ChatMessage`] (optional)
-          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is a
-          map of strings and should only be used with other chat query fields.
+          The messages field used ONLY for __chat external & foundation model__ serving endpoints. This is an
+          array of ChatMessage objects and should only be used with other chat query fields.
         :param n: int (optional)
           The n (number of candidates) field used ONLY for __completions__ and __chat external & foundation
           model__ serving endpoints. This is an integer between 1 and 5 with a default of 1 and should only be

databricks/sdk/service/sharing.py

@@ -727,6 +727,9 @@ class ListProviderShareAssetsResponse:
     notebooks: Optional[List[NotebookFile]] = None
     """The list of notebooks in the share."""
 
+    share: Optional[Share] = None
+    """The metadata of the share."""
+
     tables: Optional[List[Table]] = None
     """The list of tables in the share."""
 
@@ -740,6 +743,8 @@ class ListProviderShareAssetsResponse:
             body["functions"] = [v.as_dict() for v in self.functions]
         if self.notebooks:
             body["notebooks"] = [v.as_dict() for v in self.notebooks]
+        if self.share:
+            body["share"] = self.share.as_dict()
         if self.tables:
             body["tables"] = [v.as_dict() for v in self.tables]
         if self.volumes:
@@ -753,6 +758,8 @@ class ListProviderShareAssetsResponse:
             body["functions"] = self.functions
         if self.notebooks:
             body["notebooks"] = self.notebooks
+        if self.share:
+            body["share"] = self.share
         if self.tables:
             body["tables"] = self.tables
         if self.volumes:
@@ -765,6 +772,7 @@ class ListProviderShareAssetsResponse:
         return cls(
             functions=_repeated_dict(d, "functions", DeltaSharingFunction),
             notebooks=_repeated_dict(d, "notebooks", NotebookFile),
+            share=_from_dict(d, "share", Share),
             tables=_repeated_dict(d, "tables", Table),
             volumes=_repeated_dict(d, "volumes", Volume),
         )
@@ -1817,6 +1825,63 @@ class SecurablePropertiesKvPairs:
         return cls(properties=d.get("properties", None))
 
 
+@dataclass
+class Share:
+    comment: Optional[str] = None
+    """The comment of the share."""
+
+    display_name: Optional[str] = None
+    """The display name of the share. If defined, it will be shown in the UI."""
+
+    id: Optional[str] = None
+
+    name: Optional[str] = None
+
+    tags: Optional[List[catalog.TagKeyValue]] = None
+    """The tags of the share."""
+
+    def as_dict(self) -> dict:
+        """Serializes the Share into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.comment is not None:
+            body["comment"] = self.comment
+        if self.display_name is not None:
+            body["display_name"] = self.display_name
+        if self.id is not None:
+            body["id"] = self.id
+        if self.name is not None:
+            body["name"] = self.name
+        if self.tags:
+            body["tags"] = [v.as_dict() for v in self.tags]
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the Share into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.comment is not None:
+            body["comment"] = self.comment
+        if self.display_name is not None:
+            body["display_name"] = self.display_name
+        if self.id is not None:
+            body["id"] = self.id
+        if self.name is not None:
+            body["name"] = self.name
+        if self.tags:
+            body["tags"] = self.tags
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> Share:
+        """Deserializes the Share from a dictionary."""
+        return cls(
+            comment=d.get("comment", None),
+            display_name=d.get("display_name", None),
+            id=d.get("id", None),
+            name=d.get("name", None),
+            tags=_repeated_dict(d, "tags", catalog.TagKeyValue),
+        )
+
+
 @dataclass
 class ShareInfo:
     comment: Optional[str] = None

databricks/sdk/service/sql.py

@@ -5017,6 +5017,10 @@ class QueryMetrics:
     projected_remaining_task_total_time_ms: Optional[int] = None
     """projected remaining work to be done aggregated across all stages in the query, in milliseconds"""
 
+    projected_remaining_wallclock_time_ms: Optional[int] = None
+    """projected lower bound on remaining total task time based on
+    projected_remaining_task_total_time_ms / maximum concurrency"""
+
     provisioning_queue_start_timestamp: Optional[int] = None
     """Timestamp of when the query was enqueued waiting for a cluster to be provisioned for the
     warehouse. This field is optional and will not appear if the query skipped the provisioning
@@ -5102,6 +5106,8 @@ class QueryMetrics:
             body["photon_total_time_ms"] = self.photon_total_time_ms
         if self.projected_remaining_task_total_time_ms is not None:
             body["projected_remaining_task_total_time_ms"] = self.projected_remaining_task_total_time_ms
+        if self.projected_remaining_wallclock_time_ms is not None:
+            body["projected_remaining_wallclock_time_ms"] = self.projected_remaining_wallclock_time_ms
         if self.provisioning_queue_start_timestamp is not None:
             body["provisioning_queue_start_timestamp"] = self.provisioning_queue_start_timestamp
         if self.pruned_bytes is not None:
@@ -5161,6 +5167,8 @@ class QueryMetrics:
             body["photon_total_time_ms"] = self.photon_total_time_ms
         if self.projected_remaining_task_total_time_ms is not None:
             body["projected_remaining_task_total_time_ms"] = self.projected_remaining_task_total_time_ms
+        if self.projected_remaining_wallclock_time_ms is not None:
+            body["projected_remaining_wallclock_time_ms"] = self.projected_remaining_wallclock_time_ms
         if self.provisioning_queue_start_timestamp is not None:
             body["provisioning_queue_start_timestamp"] = self.provisioning_queue_start_timestamp
         if self.pruned_bytes is not None:
@@ -5215,6 +5223,7 @@ class QueryMetrics:
             overloading_queue_start_timestamp=d.get("overloading_queue_start_timestamp", None),
             photon_total_time_ms=d.get("photon_total_time_ms", None),
             projected_remaining_task_total_time_ms=d.get("projected_remaining_task_total_time_ms", None),
+            projected_remaining_wallclock_time_ms=d.get("projected_remaining_wallclock_time_ms", None),
             provisioning_queue_start_timestamp=d.get("provisioning_queue_start_timestamp", None),
             pruned_bytes=d.get("pruned_bytes", None),
             pruned_files_count=d.get("pruned_files_count", None),

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = "0.60.0"
+__version__ = "0.62.0"

{databricks_sdk-0.60.0.dist-info → databricks_sdk-0.62.0.dist-info}/METADATA

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