databricks-sdk 0.49.0__tar.gz → 0.50.0__tar.gz

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/CHANGELOG.md

@@ -1,5 +1,36 @@
 # Version changelog
 
+## Release v0.50.0
+
+### API Changes
+* Added [w.enable_export_notebook](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/settings/settings/enable_export_notebook.html) workspace-level service, [w.enable_notebook_table_clipboard](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/settings/settings/enable_notebook_table_clipboard.html) workspace-level service and [w.enable_results_downloading](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/settings/settings/enable_results_downloading.html) workspace-level service.
+* Added `get_credentials_for_trace_data_download()` and `get_credentials_for_trace_data_upload()` methods for [w.experiments](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/ml/experiments.html) workspace-level service.
+* Added `get_download_full_query_result()` method for [w.genie](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/dashboards/genie.html) workspace-level service.
+* Added `get_published_dashboard_token_info()` method for [w.lakeview_embedded](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/dashboards/lakeview_embedded.html) workspace-level service.
+* Added `binding_workspace_ids` field for `databricks.sdk.service.billing.BudgetPolicy`.
+* Added `download_id` field for `databricks.sdk.service.dashboards.GenieGenerateDownloadFullQueryResultResponse`.
+* Added `dashboard_output` field for `databricks.sdk.service.jobs.RunOutput`.
+* Added `dashboard_task` and `power_bi_task` fields for `databricks.sdk.service.jobs.RunTask`.
+* Added `dashboard_task` and `power_bi_task` fields for `databricks.sdk.service.jobs.SubmitTask`.
+* Added `dashboard_task` and `power_bi_task` fields for `databricks.sdk.service.jobs.Task`.
+* Added `include_features` field for `databricks.sdk.service.ml.CreateForecastingExperimentRequest`.
+* Added `models` field for `databricks.sdk.service.ml.LogInputs`.
+* Added `dataset_digest`, `dataset_name` and `model_id` fields for `databricks.sdk.service.ml.LogMetric`.
+* Added `dataset_digest`, `dataset_name`, `model_id` and `run_id` fields for `databricks.sdk.service.ml.Metric`.
+* Added `model_inputs` field for `databricks.sdk.service.ml.RunInputs`.
+* Added `client_application` field for `databricks.sdk.service.sql.QueryInfo`.
+* Added `geography` and `geometry` enum values for `databricks.sdk.service.catalog.ColumnTypeName`.
+* Added `allocation_timeout_no_healthy_and_warmed_up_clusters`, `docker_container_creation_exception`, `docker_image_too_large_for_instance_exception` and `docker_invalid_os_exception` enum values for `databricks.sdk.service.compute.TerminationReasonCode`.
+* Added `standard` enum value for `databricks.sdk.service.jobs.PerformanceTarget`.
+* Added `can_view` enum value for `databricks.sdk.service.sql.WarehousePermissionLevel`.
+* [Breaking] Changed `generate_download_full_query_result()` method for [w.genie](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/dashboards/genie.html) workspace-level service . Method path has changed.
+* [Breaking] Changed waiter for [CommandExecutionAPI.create](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/compute/command_execution.html#databricks.sdk.service.compute.CommandExecutionAPI.create) method.
+* [Breaking] Changed waiter for [CommandExecutionAPI.execute](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/compute/command_execution.html#databricks.sdk.service.compute.CommandExecutionAPI.execute) method.
+* [Breaking] Removed `error`, `status` and `transient_statement_id` fields for `databricks.sdk.service.dashboards.GenieGenerateDownloadFullQueryResultResponse`.
+* [Breaking] Removed `balanced` and `cost_optimized` enum values for `databricks.sdk.service.jobs.PerformanceTarget`.
+* [Breaking] Removed [PipelinesAPI.wait_get_pipeline_running](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/pipelines/pipelines.html#databricks.sdk.service.pipelines.PipelinesAPI.wait_get_pipeline_running) method.
+
+
 ## Release v0.49.0
 
 ### API Changes

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/PKG-INFO

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

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/databricks/sdk/__init__.py

@@ -86,9 +86,11 @@ from databricks.sdk.service.settings import (
     AibiDashboardEmbeddingApprovedDomainsAPI, AutomaticClusterUpdateAPI,
     ComplianceSecurityProfileAPI, CredentialsManagerAPI,
     CspEnablementAccountAPI, DefaultNamespaceAPI, DisableLegacyAccessAPI,
-    DisableLegacyDbfsAPI, DisableLegacyFeaturesAPI, EnableIpAccessListsAPI,
-    EnhancedSecurityMonitoringAPI, EsmEnablementAccountAPI, IpAccessListsAPI,
-    NetworkConnectivityAPI, NotificationDestinationsAPI, PersonalComputeAPI,
+    DisableLegacyDbfsAPI, DisableLegacyFeaturesAPI, EnableExportNotebookAPI,
+    EnableIpAccessListsAPI, EnableNotebookTableClipboardAPI,
+    EnableResultsDownloadingAPI, EnhancedSecurityMonitoringAPI,
+    EsmEnablementAccountAPI, IpAccessListsAPI, NetworkConnectivityAPI,
+    NotificationDestinationsAPI, PersonalComputeAPI,
     RestrictWorkspaceAdminsAPI, SettingsAPI, TokenManagementAPI, TokensAPI,
     WorkspaceConfAPI)
 from databricks.sdk.service.sharing import (ProvidersAPI,

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/databricks/sdk/service/billing.py

@@ -364,6 +364,10 @@ class BudgetConfigurationFilterWorkspaceIdClause:
 class BudgetPolicy:
     """Contains the BudgetPolicy details."""
 
+    binding_workspace_ids: Optional[List[int]] = None
+    """List of workspaces that this budget policy will be exclusively bound to. An empty binding
+    implies that this budget policy is open to any workspace in the account."""
+
     custom_tags: Optional[List[compute.CustomPolicyTag]] = None
     """A list of tags defined by the customer. At most 20 entries are allowed per policy."""
 
@@ -378,6 +382,8 @@ class BudgetPolicy:
     def as_dict(self) -> dict:
         """Serializes the BudgetPolicy into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.binding_workspace_ids:
+            body["binding_workspace_ids"] = [v for v in self.binding_workspace_ids]
         if self.custom_tags:
             body["custom_tags"] = [v.as_dict() for v in self.custom_tags]
         if self.policy_id is not None:
@@ -389,6 +395,8 @@ class BudgetPolicy:
     def as_shallow_dict(self) -> dict:
         """Serializes the BudgetPolicy into a shallow dictionary of its immediate attributes."""
         body = {}
+        if self.binding_workspace_ids:
+            body["binding_workspace_ids"] = self.binding_workspace_ids
         if self.custom_tags:
             body["custom_tags"] = self.custom_tags
         if self.policy_id is not None:
@@ -401,6 +409,7 @@ class BudgetPolicy:
     def from_dict(cls, d: Dict[str, Any]) -> BudgetPolicy:
         """Deserializes the BudgetPolicy from a dictionary."""
         return cls(
+            binding_workspace_ids=d.get("binding_workspace_ids", None),
             custom_tags=_repeated_dict(d, "custom_tags", compute.CustomPolicyTag),
             policy_id=d.get("policy_id", None),
             policy_name=d.get("policy_name", None),

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/databricks/sdk/service/catalog.py

@@ -1271,6 +1271,8 @@ class ColumnTypeName(Enum):
     DECIMAL = "DECIMAL"
     DOUBLE = "DOUBLE"
     FLOAT = "FLOAT"
+    GEOGRAPHY = "GEOGRAPHY"
+    GEOMETRY = "GEOMETRY"
     INT = "INT"
     INTERVAL = "INTERVAL"
     LONG = "LONG"
@@ -2592,6 +2594,11 @@ class CreateVolumeRequestContent:
     """The name of the volume"""
 
     volume_type: VolumeType
+    """The type of the volume. An external volume is located in the specified external location. A
+    managed volume is located in the default location which is specified by the parent schema, or
+    the parent catalog, or the Metastore. [Learn more]
+    
+    [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external"""
 
     comment: Optional[str] = None
     """The comment attached to the volume"""
@@ -9835,6 +9842,11 @@ class VolumeInfo:
     """The unique identifier of the volume"""
 
     volume_type: Optional[VolumeType] = None
+    """The type of the volume. An external volume is located in the specified external location. A
+    managed volume is located in the default location which is specified by the parent schema, or
+    the parent catalog, or the Metastore. [Learn more]
+    
+    [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external"""
 
     def as_dict(self) -> dict:
         """Serializes the VolumeInfo into a dictionary suitable for use as a JSON request body."""
@@ -9939,6 +9951,11 @@ class VolumeInfo:
 
 
 class VolumeType(Enum):
+    """The type of the volume. An external volume is located in the specified external location. A
+    managed volume is located in the default location which is specified by the parent schema, or
+    the parent catalog, or the Metastore. [Learn more]
+
+    [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external"""
 
     EXTERNAL = "EXTERNAL"
     MANAGED = "MANAGED"
@@ -14270,6 +14287,11 @@ class VolumesAPI:
         :param name: str
           The name of the volume
         :param volume_type: :class:`VolumeType`
+          The type of the volume. An external volume is located in the specified external location. A managed
+          volume is located in the default location which is specified by the parent schema, or the parent
+          catalog, or the Metastore. [Learn more]
+
+          [Learn more]: https://docs.databricks.com/aws/en/volumes/managed-vs-external
         :param comment: str (optional)
           The comment attached to the volume
         :param storage_location: str (optional)

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/databricks/sdk/service/compute.py

@@ -4788,6 +4788,10 @@ class Environment:
     Databricks), <vcs project url> E.g. dependencies: ["foo==0.0.1", "-r
     /Workspace/test/requirements.txt"]"""
 
+    jar_dependencies: Optional[List[str]] = None
+    """List of jar dependencies, should be string representing volume paths. For example:
+    `/Volumes/path/to/test.jar`."""
+
     def as_dict(self) -> dict:
         """Serializes the Environment into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -4795,6 +4799,8 @@ class Environment:
             body["client"] = self.client
         if self.dependencies:
             body["dependencies"] = [v for v in self.dependencies]
+        if self.jar_dependencies:
+            body["jar_dependencies"] = [v for v in self.jar_dependencies]
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -4804,12 +4810,18 @@ class Environment:
             body["client"] = self.client
         if self.dependencies:
             body["dependencies"] = self.dependencies
+        if self.jar_dependencies:
+            body["jar_dependencies"] = self.jar_dependencies
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> Environment:
         """Deserializes the Environment from a dictionary."""
-        return cls(client=d.get("client", None), dependencies=d.get("dependencies", None))
+        return cls(
+            client=d.get("client", None),
+            dependencies=d.get("dependencies", None),
+            jar_dependencies=d.get("jar_dependencies", None),
+        )
 
 
 @dataclass
@@ -7667,6 +7679,9 @@ class LogSyncStatus:
         return cls(last_attempted=d.get("last_attempted", None), last_exception=d.get("last_exception", None))
 
 
+MapAny = Dict[str, Any]
+
+
 @dataclass
 class MavenLibrary:
     coordinates: str
@@ -8937,6 +8952,7 @@ class TerminationReasonCode(Enum):
     ACCESS_TOKEN_FAILURE = "ACCESS_TOKEN_FAILURE"
     ALLOCATION_TIMEOUT = "ALLOCATION_TIMEOUT"
     ALLOCATION_TIMEOUT_NODE_DAEMON_NOT_READY = "ALLOCATION_TIMEOUT_NODE_DAEMON_NOT_READY"
+    ALLOCATION_TIMEOUT_NO_HEALTHY_AND_WARMED_UP_CLUSTERS = "ALLOCATION_TIMEOUT_NO_HEALTHY_AND_WARMED_UP_CLUSTERS"
     ALLOCATION_TIMEOUT_NO_HEALTHY_CLUSTERS = "ALLOCATION_TIMEOUT_NO_HEALTHY_CLUSTERS"
     ALLOCATION_TIMEOUT_NO_MATCHED_CLUSTERS = "ALLOCATION_TIMEOUT_NO_MATCHED_CLUSTERS"
     ALLOCATION_TIMEOUT_NO_READY_CLUSTERS = "ALLOCATION_TIMEOUT_NO_READY_CLUSTERS"
@@ -8989,7 +9005,10 @@ class TerminationReasonCode(Enum):
     DATA_ACCESS_CONFIG_CHANGED = "DATA_ACCESS_CONFIG_CHANGED"
     DBFS_COMPONENT_UNHEALTHY = "DBFS_COMPONENT_UNHEALTHY"
     DISASTER_RECOVERY_REPLICATION = "DISASTER_RECOVERY_REPLICATION"
+    DOCKER_CONTAINER_CREATION_EXCEPTION = "DOCKER_CONTAINER_CREATION_EXCEPTION"
     DOCKER_IMAGE_PULL_FAILURE = "DOCKER_IMAGE_PULL_FAILURE"
+    DOCKER_IMAGE_TOO_LARGE_FOR_INSTANCE_EXCEPTION = "DOCKER_IMAGE_TOO_LARGE_FOR_INSTANCE_EXCEPTION"
+    DOCKER_INVALID_OS_EXCEPTION = "DOCKER_INVALID_OS_EXCEPTION"
     DRIVER_EVICTION = "DRIVER_EVICTION"
     DRIVER_LAUNCH_TIMEOUT = "DRIVER_LAUNCH_TIMEOUT"
     DRIVER_NODE_UNREACHABLE = "DRIVER_NODE_UNREACHABLE"

{databricks_sdk-0.49.0 → databricks_sdk-0.50.0}/databricks/sdk/service/dashboards.py

@@ -21,6 +21,87 @@ from databricks.sdk.service import sql
 # all definitions in this file are in alphabetical order
 
 
+@dataclass
+class AuthorizationDetails:
+    grant_rules: Optional[List[AuthorizationDetailsGrantRule]] = None
+    """Represents downscoped permission rules with specific access rights. This field is specific to
+    `workspace_rule_set` constraint."""
+
+    resource_legacy_acl_path: Optional[str] = None
+    """The acl path of the tree store resource resource."""
+
+    resource_name: Optional[str] = None
+    """The resource name to which the authorization rule applies. This field is specific to
+    `workspace_rule_set` constraint. Format: `workspaces/{workspace_id}/dashboards/{dashboard_id}`"""
+
+    type: Optional[str] = None
+    """The type of authorization downscoping policy. Ex: `workspace_rule_set` defines access rules for
+    a specific workspace resource"""
+
+    def as_dict(self) -> dict:
+        """Serializes the AuthorizationDetails into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.grant_rules:
+            body["grant_rules"] = [v.as_dict() for v in self.grant_rules]
+        if self.resource_legacy_acl_path is not None:
+            body["resource_legacy_acl_path"] = self.resource_legacy_acl_path
+        if self.resource_name is not None:
+            body["resource_name"] = self.resource_name
+        if self.type is not None:
+            body["type"] = self.type
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the AuthorizationDetails into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.grant_rules:
+            body["grant_rules"] = self.grant_rules
+        if self.resource_legacy_acl_path is not None:
+            body["resource_legacy_acl_path"] = self.resource_legacy_acl_path
+        if self.resource_name is not None:
+            body["resource_name"] = self.resource_name
+        if self.type is not None:
+            body["type"] = self.type
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> AuthorizationDetails:
+        """Deserializes the AuthorizationDetails from a dictionary."""
+        return cls(
+            grant_rules=_repeated_dict(d, "grant_rules", AuthorizationDetailsGrantRule),
+            resource_legacy_acl_path=d.get("resource_legacy_acl_path", None),
+            resource_name=d.get("resource_name", None),
+            type=d.get("type", None),
+        )
+
+
+@dataclass
+class AuthorizationDetailsGrantRule:
+    permission_set: Optional[str] = None
+    """Permission sets for dashboard are defined in
+    iam-common/rbac-common/permission-sets/definitions/TreeStoreBasePermissionSets Ex:
+    `permissionSets/dashboard.runner`"""
+
+    def as_dict(self) -> dict:
+        """Serializes the AuthorizationDetailsGrantRule into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.permission_set is not None:
+            body["permission_set"] = self.permission_set
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the AuthorizationDetailsGrantRule into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.permission_set is not None:
+            body["permission_set"] = self.permission_set
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> AuthorizationDetailsGrantRule:
+        """Deserializes the AuthorizationDetailsGrantRule from a dictionary."""
+        return cls(permission_set=d.get("permission_set", None))
+
+
 @dataclass
 class CancelQueryExecutionResponse:
     status: Optional[List[CancelQueryExecutionResponseStatus]] = None
@@ -531,45 +612,53 @@ class GenieCreateConversationMessageRequest:
 
 @dataclass
 class GenieGenerateDownloadFullQueryResultResponse:
-    error: Optional[str] = None
-    """Error message if Genie failed to download the result"""
-
-    status: Optional[MessageStatus] = None
-    """Download result status"""
-
-    transient_statement_id: Optional[str] = None
-    """Transient Statement ID. Use this ID to track the download request in subsequent polling calls"""
+    download_id: Optional[str] = None
+    """Download ID. Use this ID to track the download request in subsequent polling calls"""
 
     def as_dict(self) -> dict:
         """Serializes the GenieGenerateDownloadFullQueryResultResponse into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.error is not None:
-            body["error"] = self.error
-        if self.status is not None:
-            body["status"] = self.status.value
-        if self.transient_statement_id is not None:
-            body["transient_statement_id"] = self.transient_statement_id
+        if self.download_id is not None:
+            body["download_id"] = self.download_id
         return body
 
     def as_shallow_dict(self) -> dict:
         """Serializes the GenieGenerateDownloadFullQueryResultResponse into a shallow dictionary of its immediate attributes."""
         body = {}
-        if self.error is not None:
-            body["error"] = self.error
-        if self.status is not None:
-            body["status"] = self.status
-        if self.transient_statement_id is not None:
-            body["transient_statement_id"] = self.transient_statement_id
+        if self.download_id is not None:
+            body["download_id"] = self.download_id
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> GenieGenerateDownloadFullQueryResultResponse:
         """Deserializes the GenieGenerateDownloadFullQueryResultResponse from a dictionary."""
-        return cls(
-            error=d.get("error", None),
-            status=_enum(d, "status", MessageStatus),
-            transient_statement_id=d.get("transient_statement_id", None),
-        )
+        return cls(download_id=d.get("download_id", None))
+
+
+@dataclass
+class GenieGetDownloadFullQueryResultResponse:
+    statement_response: Optional[sql.StatementResponse] = None
+    """SQL Statement Execution response. See [Get status, manifest, and result first
+    chunk](:method:statementexecution/getstatement) for more details."""
+
+    def as_dict(self) -> dict:
+        """Serializes the GenieGetDownloadFullQueryResultResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.statement_response:
+            body["statement_response"] = self.statement_response.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the GenieGetDownloadFullQueryResultResponse into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.statement_response:
+            body["statement_response"] = self.statement_response
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> GenieGetDownloadFullQueryResultResponse:
+        """Deserializes the GenieGetDownloadFullQueryResultResponse from a dictionary."""
+        return cls(statement_response=_from_dict(d, "statement_response", sql.StatementResponse))
 
 
 @dataclass
@@ -972,6 +1061,52 @@ class GetPublishedDashboardEmbeddedResponse:
         return cls()
 
 
+@dataclass
+class GetPublishedDashboardTokenInfoResponse:
+    authorization_details: Optional[List[AuthorizationDetails]] = None
+    """Authorization constraints for accessing the published dashboard. Currently includes
+    `workspace_rule_set` and could be enriched with `unity_catalog_privileges` before oAuth token
+    generation."""
+
+    custom_claim: Optional[str] = None
+    """Custom claim generated from external_value and external_viewer_id. Format:
+    `urn:aibi:external_data:<external_value>:<external_viewer_id>:<dashboard_id>`"""
+
+    scope: Optional[str] = None
+    """Scope defining access permissions."""
+
+    def as_dict(self) -> dict:
+        """Serializes the GetPublishedDashboardTokenInfoResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.authorization_details:
+            body["authorization_details"] = [v.as_dict() for v in self.authorization_details]
+        if self.custom_claim is not None:
+            body["custom_claim"] = self.custom_claim
+        if self.scope is not None:
+            body["scope"] = self.scope
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the GetPublishedDashboardTokenInfoResponse into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.authorization_details:
+            body["authorization_details"] = self.authorization_details
+        if self.custom_claim is not None:
+            body["custom_claim"] = self.custom_claim
+        if self.scope is not None:
+            body["scope"] = self.scope
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> GetPublishedDashboardTokenInfoResponse:
+        """Deserializes the GetPublishedDashboardTokenInfoResponse from a dictionary."""
+        return cls(
+            authorization_details=_repeated_dict(d, "authorization_details", AuthorizationDetails),
+            custom_claim=d.get("custom_claim", None),
+            scope=d.get("scope", None),
+        )
+
+
 class LifecycleState(Enum):
 
     ACTIVE = "ACTIVE"
@@ -2037,8 +2172,12 @@ class GenieAPI:
     ) -> GenieGenerateDownloadFullQueryResultResponse:
         """Generate full query result download.
 
-        Initiate full SQL query result download and obtain a transient ID for tracking the download progress.
-        This call initiates a new SQL execution to generate the query result.
+        Initiate full SQL query result download and obtain a `download_id` to track the download progress.
+        This call initiates a new SQL execution to generate the query result. The result is stored in an
+        external link can be retrieved using the [Get Download Full Query
+        Result](:method:genie/getdownloadfullqueryresult) API. Warning: Databricks strongly recommends that
+        you protect the URLs that are returned by the `EXTERNAL_LINKS` disposition. See [Execute
+        Statement](:method:statementexecution/executestatement) for more details.
 
         :param space_id: str
           Space ID
@@ -2058,11 +2197,52 @@ class GenieAPI:
 
         res = self._api.do(
             "POST",
-            f"/api/2.0/genie/spaces/{space_id}/conversations/{conversation_id}/messages/{message_id}/attachments/{attachment_id}/generate-download",
+            f"/api/2.0/genie/spaces/{space_id}/conversations/{conversation_id}/messages/{message_id}/attachments/{attachment_id}/downloads",
             headers=headers,
         )
         return GenieGenerateDownloadFullQueryResultResponse.from_dict(res)
 
+    def get_download_full_query_result(
+        self, space_id: str, conversation_id: str, message_id: str, attachment_id: str, download_id: str
+    ) -> GenieGetDownloadFullQueryResultResponse:
+        """Get download full query result.
+
+        After [Generating a Full Query Result Download](:method:genie/getdownloadfullqueryresult) and
+        successfully receiving a `download_id`, use this API to Poll download progress and retrieve the SQL
+        query result external link(s) upon completion. Warning: Databricks strongly recommends that you
+        protect the URLs that are returned by the `EXTERNAL_LINKS` disposition. When you use the
+        `EXTERNAL_LINKS` disposition, a short-lived, presigned URL is generated, which can be used to download
+        the results directly from Amazon S3. As a short-lived access credential is embedded in this presigned
+        URL, you should protect the URL. Because presigned URLs are already generated with embedded temporary
+        access credentials, you must not set an Authorization header in the download requests. See [Execute
+        Statement](:method:statementexecution/executestatement) for more details.
+
+        :param space_id: str
+          Space ID
+        :param conversation_id: str
+          Conversation ID
+        :param message_id: str
+          Message ID
+        :param attachment_id: str
+          Attachment ID
+        :param download_id: str
+          Download ID. This ID is provided by the [Generate Download
+          endpoint](:method:genie/generateDownloadFullQueryResult)
+
+        :returns: :class:`GenieGetDownloadFullQueryResultResponse`
+        """
+
+        headers = {
+            "Accept": "application/json",
+        }
+
+        res = self._api.do(
+            "GET",
+            f"/api/2.0/genie/spaces/{space_id}/conversations/{conversation_id}/messages/{message_id}/attachments/{attachment_id}/downloads/{download_id}",
+            headers=headers,
+        )
+        return GenieGetDownloadFullQueryResultResponse.from_dict(res)
+
     def get_message(self, space_id: str, conversation_id: str, message_id: str) -> GenieMessage:
         """Get conversation message.
 
@@ -2739,6 +2919,42 @@ class LakeviewEmbeddedAPI:
 
         self._api.do("GET", f"/api/2.0/lakeview/dashboards/{dashboard_id}/published/embedded", headers=headers)
 
+    def get_published_dashboard_token_info(
+        self, dashboard_id: str, *, external_value: Optional[str] = None, external_viewer_id: Optional[str] = None
+    ) -> GetPublishedDashboardTokenInfoResponse:
+        """Read an information of a published dashboard to mint an OAuth token.
+
+        Get a required authorization details and scopes of a published dashboard to mint an OAuth token. The
+        `authorization_details` can be enriched to apply additional restriction.
+
+        Example: Adding the following `authorization_details` object to downscope the viewer permission to
+        specific table ``` { type: "unity_catalog_privileges", privileges: ["SELECT"], object_type: "TABLE",
+        object_full_path: "main.default.testdata" } ```
+
+        :param dashboard_id: str
+          UUID identifying the published dashboard.
+        :param external_value: str (optional)
+          Provided external value to be included in the custom claim.
+        :param external_viewer_id: str (optional)
+          Provided external viewer id to be included in the custom claim.
+
+        :returns: :class:`GetPublishedDashboardTokenInfoResponse`
+        """
+
+        query = {}
+        if external_value is not None:
+            query["external_value"] = external_value
+        if external_viewer_id is not None:
+            query["external_viewer_id"] = external_viewer_id
+        headers = {
+            "Accept": "application/json",
+        }
+
+        res = self._api.do(
+            "GET", f"/api/2.0/lakeview/dashboards/{dashboard_id}/published/tokeninfo", query=query, headers=headers
+        )
+        return GetPublishedDashboardTokenInfoResponse.from_dict(res)
+
 
 class QueryExecutionAPI:
     """Query execution APIs for AI / BI Dashboards"""