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

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

@@ -1,5 +1,69 @@
 # Version changelog
 
+## Release v0.49.0
+
+### API Changes
+* Added `generate_download_full_query_result()` method for [w.genie](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/dashboards/genie.html) workspace-level service.
+* Added `effective_user_api_scopes`, `oauth2_app_client_id`, `oauth2_app_integration_id` and `user_api_scopes` fields for `databricks.sdk.service.apps.App`.
+* Added `abfss`, `dbfs`, `error_message`, `execution_duration_seconds`, `file`, `gcs`, `s3`, `status`, `volumes` and `workspace` fields for `databricks.sdk.service.compute.InitScriptInfoAndExecutionDetails`.
+* [Breaking] Added `forecast_granularity` field for `databricks.sdk.service.ml.CreateForecastingExperimentRequest`.
+* Added `jwks_uri` field for `databricks.sdk.service.oauth2.OidcFederationPolicy`.
+* Added `event_log` field for `databricks.sdk.service.pipelines.CreatePipeline`.
+* Added `event_log` field for `databricks.sdk.service.pipelines.EditPipeline`.
+* Added `event_log` field for `databricks.sdk.service.pipelines.PipelineSpec`.
+* Added `fallback_config` field for `databricks.sdk.service.serving.AiGatewayConfig`.
+* Added `custom_provider_config` field for `databricks.sdk.service.serving.ExternalModel`.
+* Added `fallback_config` field for `databricks.sdk.service.serving.PutAiGatewayRequest`.
+* Added `fallback_config` field for `databricks.sdk.service.serving.PutAiGatewayResponse`.
+* Added `aliases`, `comment`, `data_type`, `dependency_list`, `full_data_type`, `id`, `input_params`, `name`, `properties`, `routine_definition`, `schema`, `securable_kind`, `share`, `share_id`, `storage_location` and `tags` fields for `databricks.sdk.service.sharing.DeltaSharingFunction`.
+* Added `query_source` field for `databricks.sdk.service.sql.QueryInfo`.
+* Added `foreign_catalog` enum value for `databricks.sdk.service.catalog.CatalogType`.
+* Added `browse` enum value for `databricks.sdk.service.catalog.Privilege`.
+* Added `access_token_failure`, `allocation_timeout`, `allocation_timeout_node_daemon_not_ready`, `allocation_timeout_no_healthy_clusters`, `allocation_timeout_no_matched_clusters`, `allocation_timeout_no_ready_clusters`, `allocation_timeout_no_unallocated_clusters`, `allocation_timeout_no_warmed_up_clusters`, `aws_inaccessible_kms_key_failure`, `aws_instance_profile_update_failure`, `aws_invalid_key_pair`, `aws_invalid_kms_key_state`, `aws_resource_quota_exceeded`, `azure_packed_deployment_partial_failure`, `bootstrap_timeout_due_to_misconfig`, `budget_policy_limit_enforcement_activated`, `budget_policy_resolution_failure`, `cloud_account_setup_failure`, `cloud_operation_cancelled`, `cloud_provider_instance_not_launched`, `cloud_provider_launch_failure_due_to_misconfig`, `cloud_provider_resource_stockout_due_to_misconfig`, `cluster_operation_throttled`, `cluster_operation_timeout`, `control_plane_request_failure_due_to_misconfig`, `data_access_config_changed`, `disaster_recovery_replication`, `driver_eviction`, `driver_launch_timeout`, `driver_node_unreachable`, `driver_out_of_disk`, `driver_out_of_memory`, `driver_pod_creation_failure`, `driver_unexpected_failure`, `dynamic_spark_conf_size_exceeded`, `eos_spark_image`, `executor_pod_unscheduled`, `gcp_api_rate_quota_exceeded`, `gcp_forbidden`, `gcp_iam_timeout`, `gcp_inaccessible_kms_key_failure`, `gcp_insufficient_capacity`, `gcp_ip_space_exhausted`, `gcp_kms_key_permission_denied`, `gcp_not_found`, `gcp_resource_quota_exceeded`, `gcp_service_account_access_denied`, `gcp_service_account_not_found`, `gcp_subnet_not_ready`, `gcp_trusted_image_projects_violated`, `gke_based_cluster_termination`, `init_container_not_finished`, `instance_pool_max_capacity_reached`, `instance_pool_not_found`, `instance_unreachable_due_to_misconfig`, `internal_capacity_failure`, `invalid_aws_parameter`, `invalid_instance_placement_protocol`, `invalid_worker_image_failure`, `in_penalty_box`, `lazy_allocation_timeout`, `maintenance_mode`, `netvisor_setup_timeout`, `no_matched_k8s`, `no_matched_k8s_testing_tag`, `pod_assignment_failure`, `pod_scheduling_failure`, `resource_usage_blocked`, `secret_creation_failure`, `serverless_long_running_terminated`, `spark_image_download_throttled`, `spark_image_not_found`, `ssh_bootstrap_failure`, `storage_download_failure_due_to_misconfig`, `storage_download_failure_slow`, `storage_download_failure_throttled`, `unexpected_pod_recreation`, `user_initiated_vm_termination` and `workspace_update` enum values for `databricks.sdk.service.compute.TerminationReasonCode`.
+* Added `generated_sql_query_too_long_exception` and `missing_sql_query_exception` enum values for `databricks.sdk.service.dashboards.MessageErrorType`.
+* Added `balanced` enum value for `databricks.sdk.service.jobs.PerformanceTarget`.
+* Added `listing_resource` enum value for `databricks.sdk.service.marketplace.FileParentType`.
+* Added `app` enum value for `databricks.sdk.service.marketplace.MarketplaceFileType`.
+* Added `custom` enum value for `databricks.sdk.service.serving.ExternalModelProvider`.
+* Added `arclight_multi_tenant_azure_exchange_token` and `arclight_multi_tenant_azure_exchange_token_with_user_delegation_key` enum values for `databricks.sdk.service.settings.TokenType`.
+* [Breaking] Changed `create_experiment()` method for [w.forecasting](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/ml/forecasting.html) workspace-level service with new required argument order.
+* Changed `instance_type_id` field for `databricks.sdk.service.compute.NodeInstanceType` to be required.
+* Changed `category` field for `databricks.sdk.service.compute.NodeType` to be required.
+* [Breaking] Changed `functions` field for `databricks.sdk.service.sharing.ListProviderShareAssetsResponse` to type `databricks.sdk.service.sharing.DeltaSharingFunctionList` dataclass.
+* [Breaking] Removed `execution_details` and `script` fields for `databricks.sdk.service.compute.InitScriptInfoAndExecutionDetails`.
+* [Breaking] Removed `supports_elastic_disk` field for `databricks.sdk.service.compute.NodeType`.
+* [Breaking] Removed `data_granularity_quantity` and `data_granularity_unit` fields for `databricks.sdk.service.ml.CreateForecastingExperimentRequest`.
+* [Breaking] Removed `aliases`, `comment`, `data_type`, `dependency_list`, `full_data_type`, `id`, `input_params`, `name`, `properties`, `routine_definition`, `schema`, `securable_kind`, `share`, `share_id`, `storage_location` and `tags` fields for `databricks.sdk.service.sharing.Function`.
+
+
+## Release v0.48.0
+
+### API Changes
+* Added `execution_details` and `script` fields for `databricks.sdk.service.compute.InitScriptInfoAndExecutionDetails`.
+* Added `supports_elastic_disk` field for `databricks.sdk.service.compute.NodeType`.
+* Added `data_granularity_quantity` field for `databricks.sdk.service.ml.CreateForecastingExperimentRequest`.
+* [Breaking] Added `data_granularity_unit` field for `databricks.sdk.service.ml.CreateForecastingExperimentRequest`.
+* Added `aliases`, `comment`, `data_type`, `dependency_list`, `full_data_type`, `id`, `input_params`, `name`, `properties`, `routine_definition`, `schema`, `securable_kind`, `share`, `share_id`, `storage_location` and `tags` fields for `databricks.sdk.service.sharing.Function`.
+* [Breaking] Changed `create_experiment()` method for [w.forecasting](https://databricks-sdk-py.readthedocs.io/en/latest/workspace/ml/forecasting.html) workspace-level service with new required argument order.
+* [Breaking] Changed `instance_type_id` field for `databricks.sdk.service.compute.NodeInstanceType` to no longer be required.
+* [Breaking] Changed `category` field for `databricks.sdk.service.compute.NodeType` to no longer be required.
+* [Breaking] Changed `functions` field for `databricks.sdk.service.sharing.ListProviderShareAssetsResponse` to type `databricks.sdk.service.sharing.FunctionList` dataclass.
+* [Breaking] Removed `abfss`, `dbfs`, `error_message`, `execution_duration_seconds`, `file`, `gcs`, `s3`, `status`, `volumes` and `workspace` fields for `databricks.sdk.service.compute.InitScriptInfoAndExecutionDetails`.
+* [Breaking] Removed `forecast_granularity` field for `databricks.sdk.service.ml.CreateForecastingExperimentRequest`.
+* [Breaking] Removed `jwks_uri` field for `databricks.sdk.service.oauth2.OidcFederationPolicy`.
+* [Breaking] Removed `fallback_config` field for `databricks.sdk.service.serving.AiGatewayConfig`.
+* [Breaking] Removed `custom_provider_config` field for `databricks.sdk.service.serving.ExternalModel`.
+* [Breaking] Removed `fallback_config` field for `databricks.sdk.service.serving.PutAiGatewayRequest`.
+* [Breaking] Removed `fallback_config` field for `databricks.sdk.service.serving.PutAiGatewayResponse`.
+* [Breaking] Removed `aliases`, `comment`, `data_type`, `dependency_list`, `full_data_type`, `id`, `input_params`, `name`, `properties`, `routine_definition`, `schema`, `securable_kind`, `share`, `share_id`, `storage_location` and `tags` fields for `databricks.sdk.service.sharing.DeltaSharingFunction`.
+* [Breaking] Removed `access_token_failure`, `allocation_timeout`, `allocation_timeout_node_daemon_not_ready`, `allocation_timeout_no_healthy_clusters`, `allocation_timeout_no_matched_clusters`, `allocation_timeout_no_ready_clusters`, `allocation_timeout_no_unallocated_clusters`, `allocation_timeout_no_warmed_up_clusters`, `aws_inaccessible_kms_key_failure`, `aws_instance_profile_update_failure`, `aws_invalid_key_pair`, `aws_invalid_kms_key_state`, `aws_resource_quota_exceeded`, `azure_packed_deployment_partial_failure`, `bootstrap_timeout_due_to_misconfig`, `budget_policy_limit_enforcement_activated`, `budget_policy_resolution_failure`, `cloud_account_setup_failure`, `cloud_operation_cancelled`, `cloud_provider_instance_not_launched`, `cloud_provider_launch_failure_due_to_misconfig`, `cloud_provider_resource_stockout_due_to_misconfig`, `cluster_operation_throttled`, `cluster_operation_timeout`, `control_plane_request_failure_due_to_misconfig`, `data_access_config_changed`, `disaster_recovery_replication`, `driver_eviction`, `driver_launch_timeout`, `driver_node_unreachable`, `driver_out_of_disk`, `driver_out_of_memory`, `driver_pod_creation_failure`, `driver_unexpected_failure`, `dynamic_spark_conf_size_exceeded`, `eos_spark_image`, `executor_pod_unscheduled`, `gcp_api_rate_quota_exceeded`, `gcp_forbidden`, `gcp_iam_timeout`, `gcp_inaccessible_kms_key_failure`, `gcp_insufficient_capacity`, `gcp_ip_space_exhausted`, `gcp_kms_key_permission_denied`, `gcp_not_found`, `gcp_resource_quota_exceeded`, `gcp_service_account_access_denied`, `gcp_service_account_not_found`, `gcp_subnet_not_ready`, `gcp_trusted_image_projects_violated`, `gke_based_cluster_termination`, `init_container_not_finished`, `instance_pool_max_capacity_reached`, `instance_pool_not_found`, `instance_unreachable_due_to_misconfig`, `internal_capacity_failure`, `invalid_aws_parameter`, `invalid_instance_placement_protocol`, `invalid_worker_image_failure`, `in_penalty_box`, `lazy_allocation_timeout`, `maintenance_mode`, `netvisor_setup_timeout`, `no_matched_k8s`, `no_matched_k8s_testing_tag`, `pod_assignment_failure`, `pod_scheduling_failure`, `resource_usage_blocked`, `secret_creation_failure`, `serverless_long_running_terminated`, `spark_image_download_throttled`, `spark_image_not_found`, `ssh_bootstrap_failure`, `storage_download_failure_due_to_misconfig`, `storage_download_failure_slow`, `storage_download_failure_throttled`, `unexpected_pod_recreation`, `user_initiated_vm_termination` and `workspace_update` enum values for `databricks.sdk.service.compute.TerminationReasonCode`.
+* [Breaking] Removed `generated_sql_query_too_long_exception` and `missing_sql_query_exception` enum values for `databricks.sdk.service.dashboards.MessageErrorType`.
+* [Breaking] Removed `balanced` enum value for `databricks.sdk.service.jobs.PerformanceTarget`.
+* [Breaking] Removed `listing_resource` enum value for `databricks.sdk.service.marketplace.FileParentType`.
+* [Breaking] Removed `app` enum value for `databricks.sdk.service.marketplace.MarketplaceFileType`.
+* [Breaking] Removed `custom` enum value for `databricks.sdk.service.serving.ExternalModelProvider`.
+
+
 ## Release v0.47.0
 
 ### Bug Fixes

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

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

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks/sdk/service/apps.py

@@ -50,9 +50,16 @@ class App:
 
     effective_budget_policy_id: Optional[str] = None
 
+    effective_user_api_scopes: Optional[List[str]] = None
+    """The effective api scopes granted to the user access token."""
+
     id: Optional[str] = None
     """The unique identifier of the app."""
 
+    oauth2_app_client_id: Optional[str] = None
+
+    oauth2_app_integration_id: Optional[str] = None
+
     pending_deployment: Optional[AppDeployment] = None
     """The pending deployment of the app. A deployment is considered pending when it is being prepared
     for deployment to the app compute."""
@@ -75,6 +82,8 @@ class App:
     url: Optional[str] = None
     """The URL of the app once it is deployed."""
 
+    user_api_scopes: Optional[List[str]] = None
+
     def as_dict(self) -> dict:
         """Serializes the App into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -96,10 +105,16 @@ class App:
             body["description"] = self.description
         if self.effective_budget_policy_id is not None:
             body["effective_budget_policy_id"] = self.effective_budget_policy_id
+        if self.effective_user_api_scopes:
+            body["effective_user_api_scopes"] = [v for v in self.effective_user_api_scopes]
         if self.id is not None:
             body["id"] = self.id
         if self.name is not None:
             body["name"] = self.name
+        if self.oauth2_app_client_id is not None:
+            body["oauth2_app_client_id"] = self.oauth2_app_client_id
+        if self.oauth2_app_integration_id is not None:
+            body["oauth2_app_integration_id"] = self.oauth2_app_integration_id
         if self.pending_deployment:
             body["pending_deployment"] = self.pending_deployment.as_dict()
         if self.resources:
@@ -116,6 +131,8 @@ class App:
             body["updater"] = self.updater
         if self.url is not None:
             body["url"] = self.url
+        if self.user_api_scopes:
+            body["user_api_scopes"] = [v for v in self.user_api_scopes]
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -139,10 +156,16 @@ class App:
             body["description"] = self.description
         if self.effective_budget_policy_id is not None:
             body["effective_budget_policy_id"] = self.effective_budget_policy_id
+        if self.effective_user_api_scopes:
+            body["effective_user_api_scopes"] = self.effective_user_api_scopes
         if self.id is not None:
             body["id"] = self.id
         if self.name is not None:
             body["name"] = self.name
+        if self.oauth2_app_client_id is not None:
+            body["oauth2_app_client_id"] = self.oauth2_app_client_id
+        if self.oauth2_app_integration_id is not None:
+            body["oauth2_app_integration_id"] = self.oauth2_app_integration_id
         if self.pending_deployment:
             body["pending_deployment"] = self.pending_deployment
         if self.resources:
@@ -159,6 +182,8 @@ class App:
             body["updater"] = self.updater
         if self.url is not None:
             body["url"] = self.url
+        if self.user_api_scopes:
+            body["user_api_scopes"] = self.user_api_scopes
         return body
 
     @classmethod
@@ -174,8 +199,11 @@ class App:
             default_source_code_path=d.get("default_source_code_path", None),
             description=d.get("description", None),
             effective_budget_policy_id=d.get("effective_budget_policy_id", None),
+            effective_user_api_scopes=d.get("effective_user_api_scopes", None),
             id=d.get("id", None),
             name=d.get("name", None),
+            oauth2_app_client_id=d.get("oauth2_app_client_id", None),
+            oauth2_app_integration_id=d.get("oauth2_app_integration_id", None),
             pending_deployment=_from_dict(d, "pending_deployment", AppDeployment),
             resources=_repeated_dict(d, "resources", AppResource),
             service_principal_client_id=d.get("service_principal_client_id", None),
@@ -184,6 +212,7 @@ class App:
             update_time=d.get("update_time", None),
             updater=d.get("updater", None),
             url=d.get("url", None),
+            user_api_scopes=d.get("user_api_scopes", None),
         )
 
 

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

@@ -372,7 +372,8 @@ class BudgetPolicy:
 
     policy_name: Optional[str] = None
     """The name of the policy. - Must be unique among active policies. - Can contain only characters
-    from the ISO 8859-1 (latin1) set."""
+    from the ISO 8859-1 (latin1) set. - Can't start with reserved keywords such as
+    `databricks:default-policy`."""
 
     def as_dict(self) -> dict:
         """Serializes the BudgetPolicy into a dictionary suitable for use as a JSON request body."""

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

@@ -1065,6 +1065,7 @@ class CatalogType(Enum):
     """The type of the catalog."""
 
     DELTASHARING_CATALOG = "DELTASHARING_CATALOG"
+    FOREIGN_CATALOG = "FOREIGN_CATALOG"
     MANAGED_CATALOG = "MANAGED_CATALOG"
     SYSTEM_CATALOG = "SYSTEM_CATALOG"
 
@@ -6664,6 +6665,7 @@ class Privilege(Enum):
     ACCESS = "ACCESS"
     ALL_PRIVILEGES = "ALL_PRIVILEGES"
     APPLY_TAG = "APPLY_TAG"
+    BROWSE = "BROWSE"
     CREATE = "CREATE"
     CREATE_CATALOG = "CREATE_CATALOG"
     CREATE_CONNECTION = "CREATE_CONNECTION"

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

@@ -529,6 +529,49 @@ 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"""
+
+    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
+        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
+        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),
+        )
+
+
 @dataclass
 class GenieGetMessageQueryResultResponse:
     statement_response: Optional[sql.StatementResponse] = None
@@ -1989,6 +2032,37 @@ class GenieAPI:
         )
         return GenieGetMessageQueryResultResponse.from_dict(res)
 
+    def generate_download_full_query_result(
+        self, space_id: str, conversation_id: str, message_id: str, attachment_id: str
+    ) -> 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.
+
+        :param space_id: str
+          Space ID
+        :param conversation_id: str
+          Conversation ID
+        :param message_id: str
+          Message ID
+        :param attachment_id: str
+          Attachment ID
+
+        :returns: :class:`GenieGenerateDownloadFullQueryResultResponse`
+        """
+
+        headers = {
+            "Accept": "application/json",
+        }
+
+        res = self._api.do(
+            "POST",
+            f"/api/2.0/genie/spaces/{space_id}/conversations/{conversation_id}/messages/{message_id}/attachments/{attachment_id}/generate-download",
+            headers=headers,
+        )
+        return GenieGenerateDownloadFullQueryResultResponse.from_dict(res)
+
     def get_message(self, space_id: str, conversation_id: str, message_id: str) -> GenieMessage:
         """Get conversation message.
 

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks/sdk/service/marketplace.py

@@ -1948,7 +1948,6 @@ class ListProvidersResponse:
 @dataclass
 class Listing:
     summary: ListingSummary
-    """Next Number: 26"""
 
     detail: Optional[ListingDetail] = None
 
@@ -2252,8 +2251,6 @@ class ListingStatus(Enum):
 
 @dataclass
 class ListingSummary:
-    """Next Number: 26"""
-
     name: str
 
     listing_type: ListingType

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks/sdk/service/pipelines.py

@@ -58,6 +58,9 @@ class CreatePipeline:
     edition: Optional[str] = None
     """Pipeline product edition."""
 
+    event_log: Optional[EventLogSpec] = None
+    """Event log configuration for this pipeline"""
+
     filters: Optional[Filters] = None
     """Filters on which Pipeline packages to include in the deployed graph."""
 
@@ -136,6 +139,8 @@ class CreatePipeline:
             body["dry_run"] = self.dry_run
         if self.edition is not None:
             body["edition"] = self.edition
+        if self.event_log:
+            body["event_log"] = self.event_log.as_dict()
         if self.filters:
             body["filters"] = self.filters.as_dict()
         if self.gateway_definition:
@@ -193,6 +198,8 @@ class CreatePipeline:
             body["dry_run"] = self.dry_run
         if self.edition is not None:
             body["edition"] = self.edition
+        if self.event_log:
+            body["event_log"] = self.event_log
         if self.filters:
             body["filters"] = self.filters
         if self.gateway_definition:
@@ -240,6 +247,7 @@ class CreatePipeline:
             development=d.get("development", None),
             dry_run=d.get("dry_run", None),
             edition=d.get("edition", None),
+            event_log=_from_dict(d, "event_log", EventLogSpec),
             filters=_from_dict(d, "filters", Filters),
             gateway_definition=_from_dict(d, "gateway_definition", IngestionGatewayPipelineDefinition),
             id=d.get("id", None),
@@ -427,6 +435,9 @@ class EditPipeline:
     edition: Optional[str] = None
     """Pipeline product edition."""
 
+    event_log: Optional[EventLogSpec] = None
+    """Event log configuration for this pipeline"""
+
     expected_last_modified: Optional[int] = None
     """If present, the last-modified time of the pipeline settings before the edit. If the settings
     were modified after that time, then the request will fail with a conflict."""
@@ -510,6 +521,8 @@ class EditPipeline:
             body["development"] = self.development
         if self.edition is not None:
             body["edition"] = self.edition
+        if self.event_log:
+            body["event_log"] = self.event_log.as_dict()
         if self.expected_last_modified is not None:
             body["expected_last_modified"] = self.expected_last_modified
         if self.filters:
@@ -569,6 +582,8 @@ class EditPipeline:
             body["development"] = self.development
         if self.edition is not None:
             body["edition"] = self.edition
+        if self.event_log:
+            body["event_log"] = self.event_log
         if self.expected_last_modified is not None:
             body["expected_last_modified"] = self.expected_last_modified
         if self.filters:
@@ -619,6 +634,7 @@ class EditPipeline:
             deployment=_from_dict(d, "deployment", PipelineDeployment),
             development=d.get("development", None),
             edition=d.get("edition", None),
+            event_log=_from_dict(d, "event_log", EventLogSpec),
             expected_last_modified=d.get("expected_last_modified", None),
             filters=_from_dict(d, "filters", Filters),
             gateway_definition=_from_dict(d, "gateway_definition", IngestionGatewayPipelineDefinition),
@@ -698,6 +714,47 @@ class EventLevel(Enum):
     WARN = "WARN"
 
 
+@dataclass
+class EventLogSpec:
+    """Configurable event log parameters."""
+
+    catalog: Optional[str] = None
+    """The UC catalog the event log is published under."""
+
+    name: Optional[str] = None
+    """The name the event log is published to in UC."""
+
+    schema: Optional[str] = None
+    """The UC schema the event log is published under."""
+
+    def as_dict(self) -> dict:
+        """Serializes the EventLogSpec into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.catalog is not None:
+            body["catalog"] = self.catalog
+        if self.name is not None:
+            body["name"] = self.name
+        if self.schema is not None:
+            body["schema"] = self.schema
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the EventLogSpec into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.catalog is not None:
+            body["catalog"] = self.catalog
+        if self.name is not None:
+            body["name"] = self.name
+        if self.schema is not None:
+            body["schema"] = self.schema
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> EventLogSpec:
+        """Deserializes the EventLogSpec from a dictionary."""
+        return cls(catalog=d.get("catalog", None), name=d.get("name", None), schema=d.get("schema", None))
+
+
 @dataclass
 class FileLibrary:
     path: Optional[str] = None
@@ -2205,6 +2262,9 @@ class PipelineSpec:
     edition: Optional[str] = None
     """Pipeline product edition."""
 
+    event_log: Optional[EventLogSpec] = None
+    """Event log configuration for this pipeline"""
+
     filters: Optional[Filters] = None
     """Filters on which Pipeline packages to include in the deployed graph."""
 
@@ -2271,6 +2331,8 @@ class PipelineSpec:
             body["development"] = self.development
         if self.edition is not None:
             body["edition"] = self.edition
+        if self.event_log:
+            body["event_log"] = self.event_log.as_dict()
         if self.filters:
             body["filters"] = self.filters.as_dict()
         if self.gateway_definition:
@@ -2322,6 +2384,8 @@ class PipelineSpec:
             body["development"] = self.development
         if self.edition is not None:
             body["edition"] = self.edition
+        if self.event_log:
+            body["event_log"] = self.event_log
         if self.filters:
             body["filters"] = self.filters
         if self.gateway_definition:
@@ -2365,6 +2429,7 @@ class PipelineSpec:
             deployment=_from_dict(d, "deployment", PipelineDeployment),
             development=d.get("development", None),
             edition=d.get("edition", None),
+            event_log=_from_dict(d, "event_log", EventLogSpec),
             filters=_from_dict(d, "filters", Filters),
             gateway_definition=_from_dict(d, "gateway_definition", IngestionGatewayPipelineDefinition),
             id=d.get("id", None),
@@ -3403,6 +3468,7 @@ class PipelinesAPI:
         development: Optional[bool] = None,
         dry_run: Optional[bool] = None,
         edition: Optional[str] = None,
+        event_log: Optional[EventLogSpec] = None,
         filters: Optional[Filters] = None,
         gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None,
         id: Optional[str] = None,
@@ -3447,6 +3513,8 @@ class PipelinesAPI:
         :param dry_run: bool (optional)
         :param edition: str (optional)
           Pipeline product edition.
+        :param event_log: :class:`EventLogSpec` (optional)
+          Event log configuration for this pipeline
         :param filters: :class:`Filters` (optional)
           Filters on which Pipeline packages to include in the deployed graph.
         :param gateway_definition: :class:`IngestionGatewayPipelineDefinition` (optional)
@@ -3510,6 +3578,8 @@ class PipelinesAPI:
             body["dry_run"] = dry_run
         if edition is not None:
             body["edition"] = edition
+        if event_log is not None:
+            body["event_log"] = event_log.as_dict()
         if filters is not None:
             body["filters"] = filters.as_dict()
         if gateway_definition is not None:
@@ -3903,6 +3973,7 @@ class PipelinesAPI:
         deployment: Optional[PipelineDeployment] = None,
         development: Optional[bool] = None,
         edition: Optional[str] = None,
+        event_log: Optional[EventLogSpec] = None,
         expected_last_modified: Optional[int] = None,
         filters: Optional[Filters] = None,
         gateway_definition: Optional[IngestionGatewayPipelineDefinition] = None,
@@ -3948,6 +4019,8 @@ class PipelinesAPI:
           Whether the pipeline is in Development mode. Defaults to false.
         :param edition: str (optional)
           Pipeline product edition.
+        :param event_log: :class:`EventLogSpec` (optional)
+          Event log configuration for this pipeline
         :param expected_last_modified: int (optional)
           If present, the last-modified time of the pipeline settings before the edit. If the settings were
           modified after that time, then the request will fail with a conflict.
@@ -4012,6 +4085,8 @@ class PipelinesAPI:
             body["development"] = development
         if edition is not None:
             body["edition"] = edition
+        if event_log is not None:
+            body["event_log"] = event_log.as_dict()
         if expected_last_modified is not None:
             body["expected_last_modified"] = expected_last_modified
         if filters is not None:

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks/sdk/service/serving.py

@@ -3294,6 +3294,7 @@ class ServedModelInputWorkloadSize(Enum):
 
 
 class ServedModelInputWorkloadType(Enum):
+    """Please keep this in sync with with workload types in InferenceEndpointEntities.scala"""
 
     CPU = "CPU"
     GPU_LARGE = "GPU_LARGE"
@@ -4059,6 +4060,7 @@ class ServingEndpointPermissionsRequest:
 
 
 class ServingModelWorkloadType(Enum):
+    """Please keep this in sync with with workload types in InferenceEndpointEntities.scala"""
 
     CPU = "CPU"
     GPU_LARGE = "GPU_LARGE"

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks/sdk/service/settings.py

@@ -4146,6 +4146,10 @@ class TokenType(Enum):
 
     ARCLIGHT_AZURE_EXCHANGE_TOKEN = "ARCLIGHT_AZURE_EXCHANGE_TOKEN"
     ARCLIGHT_AZURE_EXCHANGE_TOKEN_WITH_USER_DELEGATION_KEY = "ARCLIGHT_AZURE_EXCHANGE_TOKEN_WITH_USER_DELEGATION_KEY"
+    ARCLIGHT_MULTI_TENANT_AZURE_EXCHANGE_TOKEN = "ARCLIGHT_MULTI_TENANT_AZURE_EXCHANGE_TOKEN"
+    ARCLIGHT_MULTI_TENANT_AZURE_EXCHANGE_TOKEN_WITH_USER_DELEGATION_KEY = (
+        "ARCLIGHT_MULTI_TENANT_AZURE_EXCHANGE_TOKEN_WITH_USER_DELEGATION_KEY"
+    )
     AZURE_ACTIVE_DIRECTORY_TOKEN = "AZURE_ACTIVE_DIRECTORY_TOKEN"
 
 

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks/sdk/service/sql.py

@@ -3161,6 +3161,123 @@ class ExternalLink:
         )
 
 
+@dataclass
+class ExternalQuerySource:
+    alert_id: Optional[str] = None
+    """The canonical identifier for this SQL alert"""
+
+    dashboard_id: Optional[str] = None
+    """The canonical identifier for this Lakeview dashboard"""
+
+    genie_space_id: Optional[str] = None
+    """The canonical identifier for this Genie space"""
+
+    job_info: Optional[ExternalQuerySourceJobInfo] = None
+
+    legacy_dashboard_id: Optional[str] = None
+    """The canonical identifier for this legacy dashboard"""
+
+    notebook_id: Optional[str] = None
+    """The canonical identifier for this notebook"""
+
+    sql_query_id: Optional[str] = None
+    """The canonical identifier for this SQL query"""
+
+    def as_dict(self) -> dict:
+        """Serializes the ExternalQuerySource into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.alert_id is not None:
+            body["alert_id"] = self.alert_id
+        if self.dashboard_id is not None:
+            body["dashboard_id"] = self.dashboard_id
+        if self.genie_space_id is not None:
+            body["genie_space_id"] = self.genie_space_id
+        if self.job_info:
+            body["job_info"] = self.job_info.as_dict()
+        if self.legacy_dashboard_id is not None:
+            body["legacy_dashboard_id"] = self.legacy_dashboard_id
+        if self.notebook_id is not None:
+            body["notebook_id"] = self.notebook_id
+        if self.sql_query_id is not None:
+            body["sql_query_id"] = self.sql_query_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ExternalQuerySource into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.alert_id is not None:
+            body["alert_id"] = self.alert_id
+        if self.dashboard_id is not None:
+            body["dashboard_id"] = self.dashboard_id
+        if self.genie_space_id is not None:
+            body["genie_space_id"] = self.genie_space_id
+        if self.job_info:
+            body["job_info"] = self.job_info
+        if self.legacy_dashboard_id is not None:
+            body["legacy_dashboard_id"] = self.legacy_dashboard_id
+        if self.notebook_id is not None:
+            body["notebook_id"] = self.notebook_id
+        if self.sql_query_id is not None:
+            body["sql_query_id"] = self.sql_query_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> ExternalQuerySource:
+        """Deserializes the ExternalQuerySource from a dictionary."""
+        return cls(
+            alert_id=d.get("alert_id", None),
+            dashboard_id=d.get("dashboard_id", None),
+            genie_space_id=d.get("genie_space_id", None),
+            job_info=_from_dict(d, "job_info", ExternalQuerySourceJobInfo),
+            legacy_dashboard_id=d.get("legacy_dashboard_id", None),
+            notebook_id=d.get("notebook_id", None),
+            sql_query_id=d.get("sql_query_id", None),
+        )
+
+
+@dataclass
+class ExternalQuerySourceJobInfo:
+    job_id: Optional[str] = None
+    """The canonical identifier for this job."""
+
+    job_run_id: Optional[str] = None
+    """The canonical identifier of the run. This ID is unique across all runs of all jobs."""
+
+    job_task_run_id: Optional[str] = None
+    """The canonical identifier of the task run."""
+
+    def as_dict(self) -> dict:
+        """Serializes the ExternalQuerySourceJobInfo into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.job_id is not None:
+            body["job_id"] = self.job_id
+        if self.job_run_id is not None:
+            body["job_run_id"] = self.job_run_id
+        if self.job_task_run_id is not None:
+            body["job_task_run_id"] = self.job_task_run_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ExternalQuerySourceJobInfo into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.job_id is not None:
+            body["job_id"] = self.job_id
+        if self.job_run_id is not None:
+            body["job_run_id"] = self.job_run_id
+        if self.job_task_run_id is not None:
+            body["job_task_run_id"] = self.job_task_run_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> ExternalQuerySourceJobInfo:
+        """Deserializes the ExternalQuerySourceJobInfo from a dictionary."""
+        return cls(
+            job_id=d.get("job_id", None),
+            job_run_id=d.get("job_run_id", None),
+            job_task_run_id=d.get("job_task_run_id", None),
+        )
+
+
 class Format(Enum):
 
     ARROW_STREAM = "ARROW_STREAM"
@@ -5113,6 +5230,11 @@ class QueryInfo:
     query_id: Optional[str] = None
     """The query ID."""
 
+    query_source: Optional[ExternalQuerySource] = None
+    """A struct that contains key-value pairs representing Databricks entities that were involved in
+    the execution of this statement, such as jobs, notebooks, or dashboards. This field only records
+    Databricks entities."""
+
     query_start_time_ms: Optional[int] = None
     """The time the query started."""
 
@@ -5173,6 +5295,8 @@ class QueryInfo:
             body["query_end_time_ms"] = self.query_end_time_ms
         if self.query_id is not None:
             body["query_id"] = self.query_id
+        if self.query_source:
+            body["query_source"] = self.query_source.as_dict()
         if self.query_start_time_ms is not None:
             body["query_start_time_ms"] = self.query_start_time_ms
         if self.query_text is not None:
@@ -5222,6 +5346,8 @@ class QueryInfo:
             body["query_end_time_ms"] = self.query_end_time_ms
         if self.query_id is not None:
             body["query_id"] = self.query_id
+        if self.query_source:
+            body["query_source"] = self.query_source
         if self.query_start_time_ms is not None:
             body["query_start_time_ms"] = self.query_start_time_ms
         if self.query_text is not None:
@@ -5259,6 +5385,7 @@ class QueryInfo:
             plans_state=_enum(d, "plans_state", PlansState),
             query_end_time_ms=d.get("query_end_time_ms", None),
             query_id=d.get("query_id", None),
+            query_source=_from_dict(d, "query_source", ExternalQuerySource),
             query_start_time_ms=d.get("query_start_time_ms", None),
             query_text=d.get("query_text", None),
             rows_produced=d.get("rows_produced", None),

databricks_sdk-0.49.0/databricks/sdk/version.py

@@ -0,0 +1 @@
+__version__ = "0.49.0"

{databricks_sdk-0.47.0 → databricks_sdk-0.49.0}/databricks_sdk.egg-info/PKG-INFO

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

databricks_sdk-0.47.0/databricks/sdk/version.py

@@ -1 +0,0 @@
-__version__ = "0.47.0"