databricks-sdk 0.70.0__py3-none-any.whl → 0.72.0__py3-none-any.whl

databricks/sdk/service/sql.py

@@ -331,6 +331,12 @@ class AlertEvaluationState(Enum):
     UNKNOWN = "UNKNOWN"
 
 
+class AlertLifecycleState(Enum):
+
+    ACTIVE = "ACTIVE"
+    DELETED = "DELETED"
+
+
 @dataclass
 class AlertOperandColumn:
     name: Optional[str] = None
@@ -665,7 +671,7 @@ class AlertV2:
     id: Optional[str] = None
     """UUID identifying the alert."""
 
-    lifecycle_state: Optional[LifecycleState] = None
+    lifecycle_state: Optional[AlertLifecycleState] = None
     """Indicates whether the query is trashed."""
 
     owner_user_name: Optional[str] = None
@@ -776,7 +782,7 @@ class AlertV2:
             effective_run_as=_from_dict(d, "effective_run_as", AlertV2RunAs),
             evaluation=_from_dict(d, "evaluation", AlertV2Evaluation),
             id=d.get("id", None),
-            lifecycle_state=_enum(d, "lifecycle_state", LifecycleState),
+            lifecycle_state=_enum(d, "lifecycle_state", AlertLifecycleState),
             owner_user_name=d.get("owner_user_name", None),
             parent_path=d.get("parent_path", None),
             query_text=d.get("query_text", None),
@@ -1132,24 +1138,6 @@ class BaseChunkInfo:
         )
 
 
-@dataclass
-class CancelExecutionResponse:
-    def as_dict(self) -> dict:
-        """Serializes the CancelExecutionResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the CancelExecutionResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> CancelExecutionResponse:
-        """Deserializes the CancelExecutionResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class Channel:
     """Configures the channel name and DBSQL version of the warehouse. CHANNEL_NAME_CUSTOM should be
@@ -5111,7 +5099,7 @@ class QueryMetrics:
     queue."""
 
     pruned_bytes: Optional[int] = None
-    """Total number of bytes in all tables not read due to pruning"""
+    """Total number of file bytes in all tables not read due to pruning"""
 
     pruned_files_count: Optional[int] = None
     """Total number of files from all tables not read due to pruning"""
@@ -5125,6 +5113,9 @@ class QueryMetrics:
     read_cache_bytes: Optional[int] = None
     """Size of persistent data read from the cache, in bytes."""
 
+    read_files_bytes: Optional[int] = None
+    """Total number of file bytes in all tables read"""
+
     read_files_count: Optional[int] = None
     """Number of files read after pruning"""
 
@@ -5204,6 +5195,8 @@ class QueryMetrics:
             body["read_bytes"] = self.read_bytes
         if self.read_cache_bytes is not None:
             body["read_cache_bytes"] = self.read_cache_bytes
+        if self.read_files_bytes is not None:
+            body["read_files_bytes"] = self.read_files_bytes
         if self.read_files_count is not None:
             body["read_files_count"] = self.read_files_count
         if self.read_partitions_count is not None:
@@ -5265,6 +5258,8 @@ class QueryMetrics:
             body["read_bytes"] = self.read_bytes
         if self.read_cache_bytes is not None:
             body["read_cache_bytes"] = self.read_cache_bytes
+        if self.read_files_bytes is not None:
+            body["read_files_bytes"] = self.read_files_bytes
         if self.read_files_count is not None:
             body["read_files_count"] = self.read_files_count
         if self.read_partitions_count is not None:
@@ -5314,6 +5309,7 @@ class QueryMetrics:
             query_compilation_start_timestamp=d.get("query_compilation_start_timestamp", None),
             read_bytes=d.get("read_bytes", None),
             read_cache_bytes=d.get("read_cache_bytes", None),
+            read_files_bytes=d.get("read_files_bytes", None),
             read_files_count=d.get("read_files_count", None),
             read_partitions_count=d.get("read_partitions_count", None),
             read_remote_bytes=d.get("read_remote_bytes", None),
@@ -6299,6 +6295,7 @@ class TerminationReasonCode(Enum):
     DATABASE_CONNECTION_FAILURE = "DATABASE_CONNECTION_FAILURE"
     DATA_ACCESS_CONFIG_CHANGED = "DATA_ACCESS_CONFIG_CHANGED"
     DBFS_COMPONENT_UNHEALTHY = "DBFS_COMPONENT_UNHEALTHY"
+    DBR_IMAGE_RESOLUTION_FAILURE = "DBR_IMAGE_RESOLUTION_FAILURE"
     DISASTER_RECOVERY_REPLICATION = "DISASTER_RECOVERY_REPLICATION"
     DNS_RESOLUTION_ERROR = "DNS_RESOLUTION_ERROR"
     DOCKER_CONTAINER_CREATION_EXCEPTION = "DOCKER_CONTAINER_CREATION_EXCEPTION"
@@ -7399,6 +7396,7 @@ class AlertsAPI:
 
         :returns: :class:`Alert`
         """
+
         body = {}
         if alert is not None:
             body["alert"] = alert.as_dict()
@@ -7501,6 +7499,7 @@ class AlertsAPI:
 
         :returns: :class:`Alert`
         """
+
         body = {}
         if alert is not None:
             body["alert"] = alert.as_dict()
@@ -7562,6 +7561,7 @@ class AlertsLegacyAPI:
 
         :returns: :class:`LegacyAlert`
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -7661,6 +7661,7 @@ class AlertsLegacyAPI:
 
 
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -7691,6 +7692,7 @@ class AlertsV2API:
 
         :returns: :class:`AlertV2`
         """
+
         body = alert.as_dict()
         headers = {
             "Accept": "application/json",
@@ -7777,6 +7779,7 @@ class AlertsV2API:
 
         :returns: :class:`AlertV2`
         """
+
         body = alert.as_dict()
         query = {}
         if update_mask is not None:
@@ -7821,6 +7824,7 @@ class DashboardWidgetsAPI:
 
         :returns: :class:`Widget`
         """
+
         body = {}
         if dashboard_id is not None:
             body["dashboard_id"] = dashboard_id
@@ -7882,6 +7886,7 @@ class DashboardWidgetsAPI:
 
         :returns: :class:`Widget`
         """
+
         body = {}
         if dashboard_id is not None:
             body["dashboard_id"] = dashboard_id
@@ -8027,6 +8032,7 @@ class DashboardsAPI:
 
         :returns: :class:`Dashboard`
         """
+
         body = {}
         if name is not None:
             body["name"] = name
@@ -8147,6 +8153,7 @@ class DbsqlPermissionsAPI:
 
         :returns: :class:`SetResponse`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -8179,6 +8186,7 @@ class DbsqlPermissionsAPI:
 
         :returns: :class:`Success`
         """
+
         body = {}
         if new_owner is not None:
             body["new_owner"] = new_owner
@@ -8216,6 +8224,7 @@ class QueriesAPI:
 
         :returns: :class:`Query`
         """
+
         body = {}
         if auto_resolve_display_name is not None:
             body["auto_resolve_display_name"] = auto_resolve_display_name
@@ -8348,6 +8357,7 @@ class QueriesAPI:
 
         :returns: :class:`Query`
         """
+
         body = {}
         if auto_resolve_display_name is not None:
             body["auto_resolve_display_name"] = auto_resolve_display_name
@@ -8427,6 +8437,7 @@ class QueriesLegacyAPI:
 
         :returns: :class:`LegacyQuery`
         """
+
         body = {}
         if data_source_id is not None:
             body["data_source_id"] = data_source_id
@@ -8622,6 +8633,7 @@ class QueriesLegacyAPI:
 
         :returns: :class:`LegacyQuery`
         """
+
         body = {}
         if data_source_id is not None:
             body["data_source_id"] = data_source_id
@@ -8715,6 +8727,7 @@ class QueryVisualizationsAPI:
 
         :returns: :class:`Visualization`
         """
+
         body = {}
         if visualization is not None:
             body["visualization"] = visualization.as_dict()
@@ -8760,6 +8773,7 @@ class QueryVisualizationsAPI:
 
         :returns: :class:`Visualization`
         """
+
         body = {}
         if update_mask is not None:
             body["update_mask"] = update_mask
@@ -8810,6 +8824,7 @@ class QueryVisualizationsLegacyAPI:
 
         :returns: :class:`LegacyVisualization`
         """
+
         body = {}
         if description is not None:
             body["description"] = description
@@ -8851,10 +8866,10 @@ class QueryVisualizationsLegacyAPI:
 
     def update(
         self,
-        id: str,
         *,
         created_at: Optional[str] = None,
         description: Optional[str] = None,
+        id: Optional[str] = None,
         name: Optional[str] = None,
         options: Optional[Any] = None,
         query: Optional[LegacyQuery] = None,
@@ -8868,11 +8883,11 @@ class QueryVisualizationsLegacyAPI:
 
         [Learn more]: https://docs.databricks.com/en/sql/dbsql-api-latest.html
 
-        :param id: str
-          The UUID for this visualization.
         :param created_at: str (optional)
         :param description: str (optional)
           A short description of this visualization. This is not displayed in the UI.
+        :param id: str (optional)
+          The UUID for this visualization.
         :param name: str (optional)
           The name of the visualization that appears on dashboards and the query screen.
         :param options: Any (optional)
@@ -8885,11 +8900,14 @@ class QueryVisualizationsLegacyAPI:
 
         :returns: :class:`LegacyVisualization`
         """
+
         body = {}
         if created_at is not None:
             body["created_at"] = created_at
         if description is not None:
             body["description"] = description
+        if id is not None:
+            body["id"] = id
         if name is not None:
             body["name"] = name
         if options is not None:
@@ -9223,6 +9241,7 @@ class StatementExecutionAPI:
 
         :returns: :class:`StatementResponse`
         """
+
         body = {}
         if byte_limit is not None:
             body["byte_limit"] = byte_limit
@@ -9451,6 +9470,7 @@ class WarehousesAPI:
           Long-running operation waiter for :class:`GetWarehouseResponse`.
           See :method:wait_get_warehouse_running for more details.
         """
+
         body = {}
         if auto_stop_mins is not None:
             body["auto_stop_mins"] = auto_stop_mins
@@ -9620,6 +9640,7 @@ class WarehousesAPI:
           Long-running operation waiter for :class:`GetWarehouseResponse`.
           See :method:wait_get_warehouse_running for more details.
         """
+
         body = {}
         if auto_stop_mins is not None:
             body["auto_stop_mins"] = auto_stop_mins
@@ -9806,6 +9827,7 @@ class WarehousesAPI:
 
         :returns: :class:`WarehousePermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -9861,6 +9883,7 @@ class WarehousesAPI:
 
 
         """
+
         body = {}
         if channel is not None:
             body["channel"] = channel.as_dict()
@@ -9943,6 +9966,7 @@ class WarehousesAPI:
 
         :returns: :class:`WarehousePermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]

databricks/sdk/service/tags.py

@@ -149,6 +149,7 @@ class TagPoliciesAPI:
 
         :returns: :class:`TagPolicy`
         """
+
         body = tag_policy.as_dict()
         headers = {
             "Accept": "application/json",
@@ -238,6 +239,7 @@ class TagPoliciesAPI:
 
         :returns: :class:`TagPolicy`
         """
+
         body = tag_policy.as_dict()
         query = {}
         if update_mask is not None:

databricks/sdk/service/vectorsearch.py

@@ -598,6 +598,8 @@ class EndpointStatusState(Enum):
     OFFLINE = "OFFLINE"
     ONLINE = "ONLINE"
     PROVISIONING = "PROVISIONING"
+    RED_STATE = "RED_STATE"
+    YELLOW_STATE = "YELLOW_STATE"
 
 
 class EndpointType(Enum):
@@ -1429,6 +1431,7 @@ class VectorSearchEndpointsAPI:
           Long-running operation waiter for :class:`EndpointInfo`.
           See :method:wait_get_endpoint_vector_search_endpoint_online for more details.
         """
+
         body = {}
         if budget_policy_id is not None:
             body["budget_policy_id"] = budget_policy_id
@@ -1529,6 +1532,7 @@ class VectorSearchEndpointsAPI:
 
         :returns: :class:`PatchEndpointBudgetPolicyResponse`
         """
+
         body = {}
         if budget_policy_id is not None:
             body["budget_policy_id"] = budget_policy_id
@@ -1554,6 +1558,7 @@ class VectorSearchEndpointsAPI:
 
         :returns: :class:`UpdateEndpointCustomTagsResponse`
         """
+
         body = {}
         if custom_tags is not None:
             body["custom_tags"] = [v.as_dict() for v in custom_tags]
@@ -1606,6 +1611,7 @@ class VectorSearchIndexesAPI:
 
         :returns: :class:`VectorIndex`
         """
+
         body = {}
         if delta_sync_index_spec is not None:
             body["delta_sync_index_spec"] = delta_sync_index_spec.as_dict()
@@ -1752,7 +1758,7 @@ class VectorSearchIndexesAPI:
         :param query_text: str (optional)
           Query text. Required for Delta Sync Index using model endpoint.
         :param query_type: str (optional)
-          The query type to use. Choices are `ANN` and `HYBRID`. Defaults to `ANN`.
+          The query type to use. Choices are `ANN` and `HYBRID` and `FULL_TEXT`. Defaults to `ANN`.
         :param query_vector: List[float] (optional)
           Query vector. Required for Direct Vector Access Index and Delta Sync Index using self-managed
           vectors.
@@ -1762,6 +1768,7 @@ class VectorSearchIndexesAPI:
 
         :returns: :class:`QueryVectorIndexResponse`
         """
+
         body = {}
         if columns is not None:
             body["columns"] = [v for v in columns]
@@ -1804,6 +1811,7 @@ class VectorSearchIndexesAPI:
 
         :returns: :class:`QueryVectorIndexResponse`
         """
+
         body = {}
         if endpoint_name is not None:
             body["endpoint_name"] = endpoint_name
@@ -1834,6 +1842,7 @@ class VectorSearchIndexesAPI:
 
         :returns: :class:`ScanVectorIndexResponse`
         """
+
         body = {}
         if last_primary_key is not None:
             body["last_primary_key"] = last_primary_key
@@ -1872,6 +1881,7 @@ class VectorSearchIndexesAPI:
 
         :returns: :class:`UpsertDataVectorIndexResponse`
         """
+
         body = {}
         if inputs_json is not None:
             body["inputs_json"] = inputs_json

databricks/sdk/service/workspace.py

@@ -240,24 +240,6 @@ class CreateRepoResponse:
         )
 
 
-@dataclass
-class CreateScopeResponse:
-    def as_dict(self) -> dict:
-        """Serializes the CreateScopeResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the CreateScopeResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> CreateScopeResponse:
-        """Deserializes the CreateScopeResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class CredentialInfo:
     credential_id: int
@@ -331,24 +313,6 @@ class CredentialInfo:
         )
 
 
-@dataclass
-class DeleteAclResponse:
-    def as_dict(self) -> dict:
-        """Serializes the DeleteAclResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the DeleteAclResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> DeleteAclResponse:
-        """Deserializes the DeleteAclResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class DeleteCredentialsResponse:
     def as_dict(self) -> dict:
@@ -403,24 +367,6 @@ class DeleteResponse:
         return cls()
 
 
-@dataclass
-class DeleteScopeResponse:
-    def as_dict(self) -> dict:
-        """Serializes the DeleteScopeResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the DeleteScopeResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> DeleteScopeResponse:
-        """Deserializes the DeleteScopeResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class DeleteSecretResponse:
     def as_dict(self) -> dict:
@@ -1032,42 +978,6 @@ class ObjectType(Enum):
     REPO = "REPO"
 
 
-@dataclass
-class PutAclResponse:
-    def as_dict(self) -> dict:
-        """Serializes the PutAclResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the PutAclResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> PutAclResponse:
-        """Deserializes the PutAclResponse from a dictionary."""
-        return cls()
-
-
-@dataclass
-class PutSecretResponse:
-    def as_dict(self) -> dict:
-        """Serializes the PutSecretResponse into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the PutSecretResponse into a shallow dictionary of its immediate attributes."""
-        body = {}
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> PutSecretResponse:
-        """Deserializes the PutSecretResponse from a dictionary."""
-        return cls()
-
-
 @dataclass
 class RepoAccessControlRequest:
     group_name: Optional[str] = None
@@ -1840,6 +1750,7 @@ class GitCredentialsAPI:
 
         :returns: :class:`CreateCredentialsResponse`
         """
+
         body = {}
         if git_email is not None:
             body["git_email"] = git_email
@@ -1948,6 +1859,7 @@ class GitCredentialsAPI:
 
 
         """
+
         body = {}
         if git_email is not None:
             body["git_email"] = git_email
@@ -2004,6 +1916,7 @@ class ReposAPI:
 
         :returns: :class:`CreateRepoResponse`
         """
+
         body = {}
         if path is not None:
             body["path"] = path
@@ -2129,6 +2042,7 @@ class ReposAPI:
 
         :returns: :class:`RepoPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -2165,6 +2079,7 @@ class ReposAPI:
 
 
         """
+
         body = {}
         if branch is not None:
             body["branch"] = branch
@@ -2190,6 +2105,7 @@ class ReposAPI:
 
         :returns: :class:`RepoPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -2268,6 +2184,7 @@ class SecretsAPI:
 
 
         """
+
         body = {}
         if backend_azure_keyvault is not None:
             body["backend_azure_keyvault"] = backend_azure_keyvault.as_dict()
@@ -2305,6 +2222,7 @@ class SecretsAPI:
 
 
         """
+
         body = {}
         if principal is not None:
             body["principal"] = principal
@@ -2334,6 +2252,7 @@ class SecretsAPI:
 
 
         """
+
         body = {}
         if scope is not None:
             body["scope"] = scope
@@ -2365,6 +2284,7 @@ class SecretsAPI:
 
 
         """
+
         body = {}
         if key is not None:
             body["key"] = key
@@ -2586,6 +2506,7 @@ class SecretsAPI:
 
 
         """
+
         body = {}
         if permission is not None:
             body["permission"] = permission.value
@@ -2637,6 +2558,7 @@ class SecretsAPI:
 
 
         """
+
         body = {}
         if bytes_value is not None:
             body["bytes_value"] = bytes_value
@@ -2679,6 +2601,7 @@ class WorkspaceAPI:
 
 
         """
+
         body = {}
         if path is not None:
             body["path"] = path
@@ -2807,7 +2730,7 @@ class WorkspaceAPI:
         If `path` already exists and `overwrite` is set to `false`, this call returns an error
         `RESOURCE_ALREADY_EXISTS`. To import a directory, you can use either the `DBC` format or the `SOURCE`
         format with the `language` field unset. To import a single file as `SOURCE`, you must set the
-        `language` field.
+        `language` field. Zip files within directories are not supported.
 
         :param path: str
           The absolute path of the object or directory. Importing a directory is only supported for the `DBC`
@@ -2836,6 +2759,7 @@ class WorkspaceAPI:
 
 
         """
+
         body = {}
         if content is not None:
             body["content"] = content
@@ -2893,6 +2817,7 @@ class WorkspaceAPI:
 
 
         """
+
         body = {}
         if path is not None:
             body["path"] = path
@@ -2922,6 +2847,7 @@ class WorkspaceAPI:
 
         :returns: :class:`WorkspaceObjectPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]
@@ -2953,6 +2879,7 @@ class WorkspaceAPI:
 
         :returns: :class:`WorkspaceObjectPermissions`
         """
+
         body = {}
         if access_control_list is not None:
             body["access_control_list"] = [v.as_dict() for v in access_control_list]

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = "0.70.0"
+__version__ = "0.72.0"

{databricks_sdk-0.70.0.dist-info → databricks_sdk-0.72.0.dist-info}/METADATA

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